fix test

Author: ericj-db

tests/unit/relation_configs/test_incremental_config.py

@@ -1,6 +1,7 @@
 from agate import Table
 
 from dbt.adapters.databricks.relation_configs.column_comments import ColumnCommentsConfig
+from dbt.adapters.databricks.relation_configs.column_mask import ColumnMaskConfig
 from dbt.adapters.databricks.relation_configs.comment import CommentConfig
 from dbt.adapters.databricks.relation_configs.constraints import (
     CheckConstraint,
@@ -69,6 +70,10 @@ def test_from_results(self):
                     "to_column",
                 ],
             ),
+            "column_masks": Table(
+                rows=[["col1", "mask1"], ["col2", "mask2"]],
+                column_names=["column_name", "mask_name"],
+            ),
         }
 
         config = IncrementalTableConfig.from_results(results)
@@ -121,5 +126,9 @@ def test_from_results(self):
                     ],
                     unset_constraints=[],
                 ),
+                "column_masks": ColumnMaskConfig(
+                    set_column_masks={"col1": "mask1", "col2": "mask2"},
+                    unset_column_masks=[],
+                ),
             }
         )