Simplify catalog tests

quantile-development · Nov 21, 2023 · fa57c41 · fa57c41
1 parent 11a7d64
commit fa57c41
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tests/test_elx/test_catalog.py b/tests/test_elx/test_catalog.py
@@ -116,14 +116,12 @@ def test_catalog(tap: Tap):
 def test_catalog_select(tap: Tap):
     """If we select a stream, the catalog should be updated."""
     catalog = tap.catalog.select(["animals"])
-    catalog_dict = catalog.dict(by_alias=True)
 
-    assert catalog_dict["streams"][0]["metadata"][-1]["metadata"]["selected"] == True
+    assert catalog.streams[0].is_selected == True
 
     catalog = tap.catalog.select([])
-    catalog_dict = catalog.dict(by_alias=True)
 
-    assert catalog_dict["streams"][0]["metadata"][-1]["metadata"]["selected"] == False
+    assert catalog.streams[0].is_selected == False
 
 
 def test_catalog_no_deselect(tap: Tap):
@@ -135,9 +133,8 @@ def test_catalog_no_deselect(tap: Tap):
 def test_catalog_deselect_stream(tap: Tap):
     """If we deselect a stream, the catalog should be updated."""
     catalog = tap.catalog.deselect(["animals"])
-    catalog_dict = catalog.dict(by_alias=True)
 
-    assert catalog_dict["streams"][0]["metadata"][-1]["metadata"]["selected"] == False
+    assert catalog.streams[0].is_selected == False
 
 
 def test_catalog_deselect_invalid_stream(tap: Tap):