@@ -295,8 +295,19 @@ def test_parse_type_from_json_primitive_types(self):
295295class TestTypeClassification :
296296 @pytest .mark .parametrize (
297297 "dtype" ,
298- ["string" , "varchar" , "char" , "text" , "character varying" , "character" , "nchar" , "nvarchar" ,
299- "STRING" , "VARCHAR" , "CHAR" ],
298+ [
299+ "string" ,
300+ "varchar" ,
301+ "char" ,
302+ "text" ,
303+ "character varying" ,
304+ "character" ,
305+ "nchar" ,
306+ "nvarchar" ,
307+ "STRING" ,
308+ "VARCHAR" ,
309+ "CHAR" ,
310+ ],
300311 )
301312 def test_is_string_true (self , dtype ):
302313 assert DatabricksColumn ("col" , dtype ).is_string () is True
@@ -326,9 +337,22 @@ def test_is_float_false(self, dtype):
326337
327338 @pytest .mark .parametrize (
328339 "dtype" ,
329- ["tinyint" , "smallint" , "int" , "integer" , "bigint" , "long" ,
330- "float" , "double" , "decimal" , "numeric" , "real" ,
331- "decimal(10,2)" , "decimal(38,0)" , "DECIMAL(10,2)" ],
340+ [
341+ "tinyint" ,
342+ "smallint" ,
343+ "int" ,
344+ "integer" ,
345+ "bigint" ,
346+ "long" ,
347+ "float" ,
348+ "double" ,
349+ "decimal" ,
350+ "numeric" ,
351+ "real" ,
352+ "decimal(10,2)" ,
353+ "decimal(38,0)" ,
354+ "DECIMAL(10,2)" ,
355+ ],
332356 )
333357 def test_is_number_true (self , dtype ):
334358 assert DatabricksColumn ("col" , dtype ).is_number () is True
0 commit comments