lint + install missing sqlalchemy library

databricks · Feb 13, 2024 · 4b995e0 · 4b995e0
1 parent 5a2de00
commit 4b995e0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/.github/workflows/integration.yml b/.github/workflows/integration.yml
@@ -50,7 +50,7 @@ jobs:
       #----------------------------------------------
       - name: Install dependencies
         if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true'
-        run: poetry install --no-interaction --no-root
+        run: poetry install --no-interaction --no-root --all-extras
       #----------------------------------------------
       # install your root project, if required
       #----------------------------------------------

diff --git a/src/databricks/sqlalchemy/test_local/test_parsing.py b/src/databricks/sqlalchemy/test_local/test_parsing.py
@@ -58,19 +58,15 @@ def test_extract_3l_namespace_from_constraint_string():
 
 
 def test_extract_3l_namespace_from_bad_constraint_string():
-    input = (
-        "FOREIGN KEY (`parent_user_id`) REFERENCES `pysql_dialect_compliance`.`users` (`user_id`)"
-    )
+    input = "FOREIGN KEY (`parent_user_id`) REFERENCES `pysql_dialect_compliance`.`users` (`user_id`)"
 
     with pytest.raises(DatabricksSqlAlchemyParseException):
         extract_three_level_identifier_from_constraint_string(input)
 
 
 @pytest.mark.parametrize("tschema", [None, "some_schema"])
 def test_build_fk_dict(tschema):
-    fk_constraint_string = (
-        "FOREIGN KEY (`parent_user_id`) REFERENCES `main`.`some_schema`.`users` (`user_id`)"
-    )
+    fk_constraint_string = "FOREIGN KEY (`parent_user_id`) REFERENCES `main`.`some_schema`.`users` (`user_id`)"
 
     result = build_fk_dict("some_fk_name", fk_constraint_string, schema_name=tschema)
 
@@ -127,7 +123,9 @@ def test_build_pk_dict():
     ],
 ]
 
-FMT_SAMPLE_DT_OUTPUT = [{"col_name": i[0], "data_type": i[1]} for i in RAW_SAMPLE_DTE_OUTPUT]
+FMT_SAMPLE_DT_OUTPUT = [
+    {"col_name": i[0], "data_type": i[1]} for i in RAW_SAMPLE_DTE_OUTPUT
+]
 
 
 @pytest.mark.parametrize(