From cdffd0b29d5b072639bb02118d6828f4d1b76099 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Mon, 14 Aug 2023 12:12:13 +0200 Subject: Adding upstream version 17.12.0. Signed-off-by: Daniel Baumann --- tests/test_optimizer.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'tests/test_optimizer.py') diff --git a/tests/test_optimizer.py b/tests/test_optimizer.py index 3fe53e4..a1bd309 100644 --- a/tests/test_optimizer.py +++ b/tests/test_optimizer.py @@ -277,6 +277,20 @@ class TestOptimizer(unittest.TestCase): self.assertEqual(exp.true(), optimizer.simplify.simplify(expression)) self.assertEqual(exp.true(), optimizer.simplify.simplify(expression.this)) + # CONCAT in (e.g.) Presto is parsed as Concat instead of SafeConcat which is the default type + # This test checks that simplify_concat preserves the corresponding expression types. + concat = parse_one("CONCAT('a', x, 'b', 'c')", read="presto") + simplified_concat = optimizer.simplify.simplify(concat) + + safe_concat = parse_one("CONCAT('a', x, 'b', 'c')") + simplified_safe_concat = optimizer.simplify.simplify(safe_concat) + + self.assertIs(type(simplified_concat), exp.Concat) + self.assertIs(type(simplified_safe_concat), exp.SafeConcat) + + self.assertEqual("CONCAT('a', x, 'bc')", simplified_concat.sql(dialect="presto")) + self.assertEqual("CONCAT('a', x, 'bc')", simplified_safe_concat.sql()) + def test_unnest_subqueries(self): self.check_file( "unnest_subqueries", -- cgit v1.2.3