fix linting, unit test

databricks · Aug 13, 2024 · 8564317 · 8564317
1 parent 98602d4
commit 8564317
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/functional/adapter/incremental/fixtures.py b/tests/functional/adapter/incremental/fixtures.py
@@ -1,6 +1,3 @@
-from dbt.adapters.databricks.relation_configs import tblproperties
-
-
 merge_update_columns_sql = """
 {{ config(
     materialized = 'incremental',

diff --git a/tests/unit/relation_configs/test_incremental_config.py b/tests/unit/relation_configs/test_incremental_config.py
@@ -2,6 +2,7 @@
 
 from dbt.adapters.databricks.relation_configs.incremental import IncrementalTableConfig
 from dbt.adapters.databricks.relation_configs.tags import TagsConfig
+from dbt.adapters.databricks.relation_configs.tblproperties import TblPropertiesConfig
 
 
 class TestIncrementalConfig:
@@ -13,13 +14,15 @@ def test_from_results(self):
                     ["tag2", "value2"],
                 ],
                 column_names=["tag_name", "tag_value"],
-            )
+            ),
+            "show_tblproperties": Table(rows=[["prop", "f1"]], column_names=["key", "value"]),
         }
 
         config = IncrementalTableConfig.from_results(results)
 
         assert config == IncrementalTableConfig(
             config={
                 "tags": TagsConfig(set_tags={"tag1": "value1", "tag2": "value2"}),
+                "tblproperties": TblPropertiesConfig(tblproperties={"prop": "f1"}),
             }
         )