diff --git a/.gitignore b/.gitignore index 15786eb..f15a193 100644 --- a/.gitignore +++ b/.gitignore @@ -2,3 +2,10 @@ __pycache__ .venv dist relationalize.egg-info +/.idea/inspectionProfiles/profiles_settings.xml +/.idea/.gitignore +/.idea/aws.xml +/.idea/misc.xml +/.idea/modules.xml +/.idea/relationalize.iml +/.idea/vcs.xml diff --git a/test/schema.test.py b/test/schema.test.py index e68f19b..46db090 100644 --- a/test/schema.test.py +++ b/test/schema.test.py @@ -1,3 +1,4 @@ +import re import unittest from copy import deepcopy @@ -16,7 +17,7 @@ CASE_5 = {"1": "foobar"} CASE_1_DDL = """ -CREATE TABLE "public"."test" ( +CREATE TABLE IF NOT EXISTS "public"."test" ( "1" BIGINT , "2" VARCHAR(65535) , "3" BOOLEAN @@ -25,7 +26,7 @@ """.strip() CASE_2_DDL = """ -CREATE TABLE "public"."test" ( +CREATE TABLE IF NOT EXISTS "public"."test" ( "1_int" BIGINT , "1_str" VARCHAR(65535) , "2_float" FLOAT @@ -121,14 +122,14 @@ def test_convert_object_choice(self): def test_generate_ddl_no_choice(self): schema1 = Schema() schema1.read_object(CASE_1) - self.assertEqual(CASE_1_DDL, schema1.generate_ddl("test")) + self.assertEqual(re.sub(r'\s+', ' ', CASE_1_DDL), re.sub(r'\s+', ' ', schema1.generate_ddl("test"))) def test_generate_ddl_choice(self): schema1 = Schema() schema1.read_object(CASE_1) schema1.read_object(CASE_2) - self.assertEqual(CASE_2_DDL, schema1.generate_ddl("test")) + self.assertEqual(re.sub(r'\s+', ' ', CASE_2_DDL), re.sub(r'\s+', ' ', schema1.generate_ddl("test"))) def test_none_cases(self): schema1 = Schema()