chore: fix test

equinor · Aug 1, 2023 · 9a5047d · 9a5047d
1 parent b99575a
commit 9a5047d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py
@@ -7,4 +7,8 @@ class ImportEntityTest(unittest.TestCase):
     def test_get_root_package_content(self):
         path = "tests/test_data/test_app_dir_struct"
         datasource_contents = get_root_packages_in_data_sources(path)
-        assert datasource_contents == {"DemoApplicationDataSource": ["models", "instances"]}
+        assert "DemoApplicationDataSource" in datasource_contents
+        assert (
+            "instances" in datasource_contents["DemoApplicationDataSource"]
+            and "models" in datasource_contents["DemoApplicationDataSource"]
+        )