minor test cleanup

dbt-labs · Apr 5, 2024 · eb940fc · eb940fc
1 parent f863015
commit eb940fc
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/functional/adapter/test_get_last_relation_modified.py b/tests/functional/adapter/test_get_last_relation_modified.py
@@ -7,7 +7,7 @@
 from dbt.cli.main import dbtRunner
 
 
-freshness_via_metadata_schema_yml = """version: 2
+freshness_via_metadata_schema_yml = """
 sources:
   - name: test_source
     freshness:
@@ -18,7 +18,7 @@
       - name: test_table
 """
 
-freshness_metadata_schema_batch_yml = """version: 2
+freshness_metadata_schema_batch_yml = """
 sources:
   - name: test_source
     freshness:
@@ -115,10 +115,11 @@ def test_get_last_relation_modified_batch(self, project, set_env_vars, custom_sc
             "test_table_with_loaded_at_field", freshness_results_batch
         )
 
+        # Remove TableLastModifiedMetadataBatch and run freshness on same input without batch strategy
         capabilities_no_batch = CapabilityDict(
             {
                 capability: support
-                for capability, support in SnowflakeAdapter._capabilities.items()
+                for capability, support in SnowflakeAdapter.capabilities().items()
                 if capability != Capability.TableLastModifiedMetadataBatch
             }
         )