@@ -181,7 +181,7 @@ def test_measurments():
181181
182182
183183def 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