Update test for list_tables

VectorInstitute · Sep 15, 2023 · a48943b · a48943b
1 parent f200277
commit a48943b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/cyclops/query/test_base.py b/tests/cyclops/query/test_base.py
@@ -10,6 +10,8 @@ def test_dataset_querier():
     """Test base querier methods using OMOPQuerier."""
     querier = OMOPQuerier("cdm_synthea10", database="synthea_integration_test")
     assert len(querier.list_tables()) == 69
+    assert len(querier.list_schemas()) == 4
+    assert len(querier.list_tables(schema_name="cdm_synthea10")) == 43
     visit_occrrence_columns = querier.list_columns("cdm_synthea10", "visit_occurrence")
     assert len(visit_occrrence_columns) == 17
     assert "visit_occurrence_id" in visit_occrrence_columns