Skip to content

Commit 92a9c7a

Browse files
committed
Add sanitize_table_name() tests #161
1 parent 5d52da4 commit 92a9c7a

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

testing/test_awswrangler/test_data_lake.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -552,7 +552,7 @@ def test_athena_read_list(database):
552552
wr.athena.read_sql_query(sql=f"SELECT ARRAY[1, 2, 3]", database=database, ctas_approach=False)
553553

554554

555-
def test_normalize_column_name():
555+
def test_sanitize_names():
556556
assert wr.catalog.sanitize_column_name("CamelCase") == "camel_case"
557557
assert wr.catalog.sanitize_column_name("CamelCase2") == "camel_case2"
558558
assert wr.catalog.sanitize_column_name("Camel_Case3") == "camel_case3"
@@ -562,6 +562,15 @@ def test_normalize_column_name():
562562
assert wr.catalog.sanitize_column_name("Camel.Case7") == "camel_case7"
563563
assert wr.catalog.sanitize_column_name("xyz_cd") == "xyz_cd"
564564
assert wr.catalog.sanitize_column_name("xyz_Cd") == "xyz_cd"
565+
assert wr.catalog.sanitize_table_name("CamelCase") == "camel_case"
566+
assert wr.catalog.sanitize_table_name("CamelCase2") == "camel_case2"
567+
assert wr.catalog.sanitize_table_name("Camel_Case3") == "camel_case3"
568+
assert wr.catalog.sanitize_table_name("Cámël_Casë4仮") == "camel_case4_"
569+
assert wr.catalog.sanitize_table_name("Camel__Case5") == "camel__case5"
570+
assert wr.catalog.sanitize_table_name("Camel{}Case6") == "camel_case6"
571+
assert wr.catalog.sanitize_table_name("Camel.Case7") == "camel_case7"
572+
assert wr.catalog.sanitize_table_name("xyz_cd") == "xyz_cd"
573+
assert wr.catalog.sanitize_table_name("xyz_Cd") == "xyz_cd"
565574

566575

567576
def test_athena_ctas_empty(database):

0 commit comments

Comments
 (0)