dbcli/sqlglot
1
0
Fork 0
Code Issues Activity

Adding upstream version 18.17.0.

Browse source 
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
Daniel Baumann 2025-02-13 21:08:47 +01:00
parent fc6bad5705
commit 03001ce1e6
Signed by: daniel
GPG key ID: FBB4F0E80A80222F
90 changed files with 46581 additions and 43319 deletions
Download patch file Download diff file Expand all files Collapse all files

41
tests/test_optimizer.py
View file

@ -550,6 +550,47 @@ FROM READ_CSV('tests/fixtures/optimizer/tpc-h/nation.csv.gz', 'delimiter', '|')
self.assertEqual(expression.right.this.left.type.this, exp.DataType.Type.INT)
self.assertEqual(expression.right.this.right.type.this, exp.DataType.Type.INT)
def test_bracket_annotation(self):
expression = annotate_types(parse_one("SELECT A[:]")).expressions[0]
self.assertEqual(expression.type.this, exp.DataType.Type.UNKNOWN)
self.assertEqual(expression.expressions[0].type.this, exp.DataType.Type.UNKNOWN)
expression = annotate_types(parse_one("SELECT ARRAY[1, 2, 3][1]")).expressions[0]
self.assertEqual(expression.this.type.sql(), "ARRAY<INT>")
self.assertEqual(expression.type.this, exp.DataType.Type.INT)
expression = annotate_types(parse_one("SELECT ARRAY[1, 2, 3][1 : 2]")).expressions[0]
self.assertEqual(expression.this.type.sql(), "ARRAY<INT>")
self.assertEqual(expression.type.sql(), "ARRAY<INT>")
expression = annotate_types(
parse_one("SELECT ARRAY[ARRAY[1], ARRAY[2], ARRAY[3]][1][2]")
).expressions[0]
self.assertEqual(expression.this.this.type.sql(), "ARRAY<ARRAY<INT>>")
self.assertEqual(expression.this.type.sql(), "ARRAY<INT>")
self.assertEqual(expression.type.this, exp.DataType.Type.INT)
expression = annotate_types(
parse_one("SELECT ARRAY[ARRAY[1], ARRAY[2], ARRAY[3]][1:2]")
).expressions[0]
self.assertEqual(expression.type.sql(), "ARRAY<ARRAY<INT>>")
expression = annotate_types(parse_one("MAP(1.0, 2, '2', 3.0)['2']", read="spark"))
self.assertEqual(expression.type.this, exp.DataType.Type.DOUBLE)
expression = annotate_types(parse_one("MAP(1.0, 2, x, 3.0)[2]", read="spark"))
self.assertEqual(expression.type.this, exp.DataType.Type.UNKNOWN)
expression = annotate_types(parse_one("MAP(ARRAY(1.0, x), ARRAY(2, 3.0))[x]"))
self.assertEqual(expression.type.this, exp.DataType.Type.DOUBLE)
expression = annotate_types(
parse_one("SELECT MAP(1.0, 2, 2, t.y)[2] FROM t", read="spark"),
schema={"t": {"y": "int"}},
).expressions[0]
self.assertEqual(expression.type.this, exp.DataType.Type.INT)
def test_interval_math_annotation(self):
schema = {
"x": {