From b737ee75da2515a4a53956e41ae85e29dd67f21d Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 6 Aug 2023 09:48:08 +0200 Subject: Adding upstream version 17.9.1. Signed-off-by: Daniel Baumann --- tests/test_schema.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) (limited to 'tests/test_schema.py') diff --git a/tests/test_schema.py b/tests/test_schema.py index b89754f..626fa11 100644 --- a/tests/test_schema.py +++ b/tests/test_schema.py @@ -238,3 +238,37 @@ class TestSchema(unittest.TestCase): schema = MappingSchema(schema={"Foo": {"`BaR`": "int"}}, dialect="bigquery") self.assertEqual(schema.column_names("Foo"), ["bar"]) self.assertEqual(schema.column_names("foo"), []) + + # Check that the schema's normalization setting can be overridden + schema = MappingSchema(schema={"X": {"y": "int"}}, normalize=False, dialect="snowflake") + self.assertEqual(schema.column_names("x", normalize=True), ["y"]) + + def test_same_number_of_qualifiers(self): + schema = MappingSchema({"x": {"y": {"c1": "int"}}}) + + with self.assertRaises(SchemaError) as ctx: + schema.add_table("z", {"c2": "int"}) + + self.assertEqual( + str(ctx.exception), + "Table z must match the schema's nesting level: 2.", + ) + + schema = MappingSchema() + schema.add_table("x.y", {"c1": "int"}) + + with self.assertRaises(SchemaError) as ctx: + schema.add_table("z", {"c2": "int"}) + + self.assertEqual( + str(ctx.exception), + "Table z must match the schema's nesting level: 2.", + ) + + with self.assertRaises(SchemaError) as ctx: + MappingSchema({"x": {"y": {"c1": "int"}}, "z": {"c2": "int"}}) + + self.assertEqual( + str(ctx.exception), + "Table z must match the schema's nesting level: 2.", + ) -- cgit v1.2.3