|
4 | 4 | import petab |
5 | 5 | from petab.parameter_mapping import _apply_parameter_table |
6 | 6 | from petab.C import * |
| 7 | +from math import nan |
7 | 8 |
|
8 | 9 | # import fixtures |
9 | 10 | pytest_plugins = [ |
@@ -268,10 +269,18 @@ def test_all_override(condition_df_2_conditions): |
268 | 269 | @staticmethod |
269 | 270 | def test_partial_override(condition_df_2_conditions): |
270 | 271 | # Condition-specific parameters, keeping original parameters |
271 | | - condition_df = condition_df_2_conditions |
| 272 | + condition_df = pd.DataFrame(data={ |
| 273 | + 'conditionId': ['condition1', 'condition2'], |
| 274 | + 'conditionName': ['', 'Condition 2'], |
| 275 | + 'fixedParameter1': [1.0, 2.0], |
| 276 | + 'fixedParameter2': [nan, 2.5], |
| 277 | + }) |
| 278 | + condition_df.set_index('conditionId', inplace=True) |
272 | 279 |
|
273 | 280 | import simplesbml |
274 | 281 | ss_model = simplesbml.SbmlModel() |
| 282 | + ss_model.addParameter('fixedParameter1', 0.5) |
| 283 | + ss_model.addParameter('fixedParameter2', 1.0) |
275 | 284 | ss_model.addParameter('dynamicParameter1', 0.0) |
276 | 285 | ss_model.addParameter('observableParameter1_obs1', 0.0) |
277 | 286 | ss_model.addParameter('observableParameter2_obs1', 0.0) |
@@ -300,26 +309,30 @@ def test_partial_override(condition_df_2_conditions): |
300 | 309 |
|
301 | 310 | expected = [({}, |
302 | 311 | {'fixedParameter1': 1.0, |
| 312 | + 'fixedParameter2': 1.0, |
303 | 313 | 'dynamicParameter1': 'dynamicParameter1', |
304 | 314 | 'observableParameter1_obs1': 'obs1par1override', |
305 | 315 | 'observableParameter2_obs1': 'obs1par2cond1override', |
306 | 316 | 'observableParameter1_obs2': np.nan, |
307 | 317 | }, |
308 | 318 | {}, |
309 | 319 | {'fixedParameter1': LIN, |
| 320 | + 'fixedParameter2': LIN, |
310 | 321 | 'dynamicParameter1': LIN, |
311 | 322 | 'observableParameter1_obs1': LIN, |
312 | 323 | 'observableParameter2_obs1': LIN, |
313 | 324 | 'observableParameter1_obs2': LIN}), |
314 | 325 | ({}, |
315 | 326 | {'fixedParameter1': 2.0, |
| 327 | + 'fixedParameter2': 2.5, |
316 | 328 | 'dynamicParameter1': 'dynamicParameter1', |
317 | 329 | 'observableParameter1_obs1': 'obs1par1override', |
318 | 330 | 'observableParameter2_obs1': 'obs1par2cond2override', |
319 | 331 | 'observableParameter1_obs2': 'obs2par1cond2override' |
320 | 332 | }, |
321 | 333 | {}, |
322 | 334 | {'fixedParameter1': LIN, |
| 335 | + 'fixedParameter2': LIN, |
323 | 336 | 'dynamicParameter1': LIN, |
324 | 337 | 'observableParameter1_obs1': LIN, |
325 | 338 | 'observableParameter2_obs1': LIN, |
|
0 commit comments