PR with da6a25d. fix unit tests

aaraney · Feb 10, 2023 · 998f191 · 998f191
1 parent 5221c65
commit 998f191
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/lib/core/dmod/test/test_meta_data.py b/python/lib/core/dmod/test/test_meta_data.py
@@ -157,7 +157,7 @@ def test_it_works(self):
         o = DataDomain(
             data_format=DataFormat.AORC_CSV,
             discrete_restrictions=[disc_rest],
-            data_fields=dict(a="str", b="float", c="int", d="datetime"),
+            custom_data_fields=dict(a="str", b="float", c="int", d="datetime"),
         )
         self.assertEqual(o.custom_data_fields["a"], str)
         self.assertEqual(o.custom_data_fields["b"], float)
@@ -207,7 +207,7 @@ def test_to_dict(self):
     def test_factory_init_from_restriction_collections(self):
         catchment_id = ["12"]
         o = DataDomain.factory_init_from_restriction_collections(data_format=DataFormat.AORC_CSV, CATCHMENT_ID=catchment_id)
-        self.assertListEqual(o.discrete_restrictions[0].values, catchment_id)
+        self.assertListEqual(o.discrete_restrictions[StandardDatasetIndex.CATCHMENT_ID].values, catchment_id)
 
     def test_factory_init_from_restriction_collections_fail_for_mismatching_index_field(self):
         with self.assertRaises(RuntimeError):