catch GxDatasourceWarning

great-expectations · Jun 6, 2024 · 5a430cd · 5a430cd
1 parent 999d0d9
commit 5a430cd
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/integration/db/taxi_data_utils.py b/tests/integration/db/taxi_data_utils.py
@@ -8,6 +8,7 @@
 
 import great_expectations as gx
 from great_expectations.core.batch_definition import BatchDefinition
+from great_expectations.datasource.fluent import GxDatasourceWarning
 from great_expectations.execution_engine.sqlalchemy_batch_data import (
     SqlAlchemyBatchData,
 )
@@ -124,6 +125,12 @@ def _execute_taxi_partitioning_test_cases(
                 message="The `schema_name argument` is deprecated",
                 category=DeprecationWarning,
             )
+            warnings.filterwarnings(
+                "once",
+                message="schema_name None does not match datasource schema ***",
+                category=GxDatasourceWarning,
+            )
+
             asset = datasource.add_table_asset(
                 data_asset_name, table_name=table_name, schema_name=None
             )