Create custom parameter equality method for test case, only generate …

…description for stratified params with description

mira/metamodel/ops.py

@@ -266,7 +266,8 @@ def stratify(
  stratum = stratified_param_tuple[0]
  stratified_param = stratified_param_tuple[1]
  d.name = stratified_param
- d.description = f"{d.description} stratified by {key}"
+ if d.description:
+ d.description = f"{d.description} stratified by {key}"
  d.with_context(inplace=True,**{key:stratum})
  parameters[stratified_param] = d
 

tests/test_ops.py

@@ -154,7 +154,25 @@ def test_stratify_full(self):
  },
  {k: i.expression for k, i in actual.initials.items()}
  )
- self.assertEqual(tm_stratified.parameters, actual.parameters)
+ def compare_parameters_no_context(tm0_params, tm1_params):
+ if tm0_params.keys() != tm1_params.keys():
+ return False
+
+ for param_id in tm0_params:
+ param_obj_0 = tm0_params[param_id]
+ param_obj_1 = tm1_params[param_id]
+
+ for param_attr in dict(param_obj_0):
+ if param_attr == "context" or param_attr == "description":
+ continue
+ if getattr(param_obj_0, param_attr) != getattr(param_obj_1, param_attr):
+ return False
+ return True
+
+ compare_parameters_no_context(tm_stratified.parameters,
+ actual.parameters)
+ self.assertTrue(compare_parameters_no_context(
+ tm_stratified.parameters, actual.parameters))
  self.assertTrue(actual.initials['infected_population_vaccinated'].expression.equals(
  tm_stratified.initials['infected_population_vaccinated'].expression))
  self.assertEqual(