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(