Skip to content

Commit dbdcd56

Browse files
committed
ensure that all observables have a noise_formula
1 parent fafaa53 commit dbdcd56

1 file changed

Lines changed: 13 additions & 5 deletions

File tree

tests/v2/test_core.py

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ def test_measurments():
181181

182182

183183
def test_observable():
184-
Observable(id="obs1", formula=x + y)
184+
Observable(id="obs1", formula=x + y, noiseFormula=1)
185185
Observable(id="obs1", formula="x + y", noise_formula="x + y")
186186
Observable(id="obs1", formula=1, noise_formula=2)
187187
Observable(
@@ -198,9 +198,17 @@ def test_observable():
198198
observable_parameters=[sp.Symbol("p1")],
199199
noise_parameters=[sp.Symbol("n1")],
200200
)
201-
assert Observable(id="obs1", formula="x + y", non_petab=1).non_petab == 1
201+
assert (
202+
Observable(
203+
id="obs1",
204+
formula="x + y",
205+
noise_formula="x + y",
206+
non_petab=1,
207+
).non_petab
208+
== 1
209+
)
202210

203-
o = Observable(id="obs1", formula=x + y)
211+
o = Observable(id="obs1", formula=x + y, noise_formula=1)
204212
assert o.observable_placeholders == []
205213
assert o.noise_placeholders == []
206214

@@ -492,14 +500,14 @@ def test_modify_problem():
492500
problem.condition_df, exp_condition_df, check_dtype=False
493501
)
494502

495-
problem.add_observable("observable1", "1")
503+
problem.add_observable("observable1", "1", noise_formula=1)
496504
problem.add_observable("observable2", "2", noise_formula=2.2)
497505

498506
exp_observable_df = pd.DataFrame(
499507
data={
500508
OBSERVABLE_ID: ["observable1", "observable2"],
501509
OBSERVABLE_FORMULA: [1, 2],
502-
NOISE_FORMULA: [np.nan, 2.2],
510+
NOISE_FORMULA: [1, 2.2],
503511
}
504512
).set_index([OBSERVABLE_ID])
505513
assert_frame_equal(

0 commit comments

Comments
 (0)