@@ -58,22 +58,26 @@ def test_extract_3l_namespace_from_constraint_string():
58
58
59
59
60
60
def test_extract_3l_namespace_from_bad_constraint_string ():
61
- input = "FOREIGN KEY (`parent_user_id`) REFERENCES `pysql_dialect_compliance`.`users` (`user_id`)"
61
+ input = (
62
+ "FOREIGN KEY (`parent_user_id`) REFERENCES `pysql_dialect_compliance`.`users` (`user_id`)"
63
+ )
62
64
63
65
with pytest .raises (DatabricksSqlAlchemyParseException ):
64
66
extract_three_level_identifier_from_constraint_string (input )
65
67
66
68
67
- @pytest .mark .parametrize ("schema" , [None , "some_schema" ])
68
- def test_build_fk_dict (schema ):
69
- fk_constraint_string = "FOREIGN KEY (`parent_user_id`) REFERENCES `main`.`some_schema`.`users` (`user_id`)"
69
+ @pytest .mark .parametrize ("tschema" , [None , "some_schema" ])
70
+ def test_build_fk_dict (tschema ):
71
+ fk_constraint_string = (
72
+ "FOREIGN KEY (`parent_user_id`) REFERENCES `main`.`some_schema`.`users` (`user_id`)"
73
+ )
70
74
71
- result = build_fk_dict ("some_fk_name" , fk_constraint_string , schema_name = schema )
75
+ result = build_fk_dict ("some_fk_name" , fk_constraint_string , schema_name = tschema )
72
76
73
77
assert result == {
74
78
"name" : "some_fk_name" ,
75
79
"constrained_columns" : ["parent_user_id" ],
76
- "referred_schema" : schema ,
80
+ "referred_schema" : tschema ,
77
81
"referred_table" : "users" ,
78
82
"referred_columns" : ["user_id" ],
79
83
}
@@ -123,9 +127,7 @@ def test_build_pk_dict():
123
127
],
124
128
]
125
129
126
- FMT_SAMPLE_DT_OUTPUT = [
127
- {"col_name" : i [0 ], "data_type" : i [1 ]} for i in RAW_SAMPLE_DTE_OUTPUT
128
- ]
130
+ FMT_SAMPLE_DT_OUTPUT = [{"col_name" : i [0 ], "data_type" : i [1 ]} for i in RAW_SAMPLE_DTE_OUTPUT ]
129
131
130
132
131
133
@pytest .mark .parametrize (
0 commit comments