testing cleaner if statement

great-expectations · Aug 12, 2021 · 7a06de4 · 7a06de4
1 parent 3580cf4
commit 7a06de4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_definitions/test_expectations_cfe.py b/tests/test_definitions/test_expectations_cfe.py
@@ -145,7 +145,8 @@ def pytest_generate_tests(metafunc):
                                     "bigquery" in test["only_for"]
                                     and BigQueryDialect is not None
                                     and hasattr(
-                                        validator_with_data.execution_engine.active_batch_data.sql_engine_dialect.BigQueryDialect
+                                        validator_with_data.execution_engine.active_batch_data.sql_engine_dialect,
+                                        "BigQueryDialect",
                                     ),
                                 ):
                                     generate_test = True
@@ -245,7 +246,8 @@ def pytest_generate_tests(metafunc):
                                     SqlAlchemyBatchData,
                                 )
                                 and hasattr(
-                                    validator_with_data.execution_engine.active_batch_data.sql_engine_dialect.BigQueryDialect
+                                    validator_with_data.execution_engine.active_batch_data.sql_engine_dialect,
+                                    "BigQueryDialect",
                                 )
                             )
                             or (