diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-06-02 23:59:40 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-06-02 23:59:46 +0000 |
commit | 20739a12c39121a9e7ad3c9a2469ec5a6876199d (patch) | |
tree | c000de91c59fd29b2d9beecf9f93b84e69727f37 /tests/test_helper.py | |
parent | Releasing debian version 12.2.0-1. (diff) | |
download | sqlglot-20739a12c39121a9e7ad3c9a2469ec5a6876199d.tar.xz sqlglot-20739a12c39121a9e7ad3c9a2469ec5a6876199d.zip |
Merging upstream version 15.0.0.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'tests/test_helper.py')
-rw-r--r-- | tests/test_helper.py | 40 |
1 files changed, 39 insertions, 1 deletions
diff --git a/tests/test_helper.py b/tests/test_helper.py index d37c03a..82d917e 100644 --- a/tests/test_helper.py +++ b/tests/test_helper.py @@ -1,6 +1,7 @@ import unittest -from sqlglot.helper import tsort +from sqlglot.dialects import BigQuery, Dialect, Snowflake +from sqlglot.helper import name_sequence, tsort class TestHelper(unittest.TestCase): @@ -29,3 +30,40 @@ class TestHelper(unittest.TestCase): "c": [], } ) + + def test_compare_dialects(self): + bigquery_class = Dialect["bigquery"] + bigquery_object = BigQuery() + bigquery_string = "bigquery" + + snowflake_class = Dialect["snowflake"] + snowflake_object = Snowflake() + snowflake_string = "snowflake" + + self.assertEqual(snowflake_class, snowflake_class) + self.assertEqual(snowflake_class, snowflake_object) + self.assertEqual(snowflake_class, snowflake_string) + self.assertEqual(snowflake_object, snowflake_object) + self.assertEqual(snowflake_object, snowflake_string) + + self.assertNotEqual(snowflake_class, bigquery_class) + self.assertNotEqual(snowflake_class, bigquery_object) + self.assertNotEqual(snowflake_class, bigquery_string) + self.assertNotEqual(snowflake_object, bigquery_object) + self.assertNotEqual(snowflake_object, bigquery_string) + + self.assertTrue(snowflake_class in {"snowflake", "bigquery"}) + self.assertTrue(snowflake_object in {"snowflake", "bigquery"}) + self.assertFalse(snowflake_class in {"bigquery", "redshift"}) + self.assertFalse(snowflake_object in {"bigquery", "redshift"}) + + def test_name_sequence(self): + s1 = name_sequence("a") + s2 = name_sequence("b") + + self.assertEqual(s1(), "a0") + self.assertEqual(s1(), "a1") + self.assertEqual(s2(), "b0") + self.assertEqual(s1(), "a2") + self.assertEqual(s2(), "b1") + self.assertEqual(s2(), "b2") |