Update types integration tests

Author: lpoulain

tests/integration/test_types_integration.py

@@ -15,16 +15,34 @@ def trino_connection(run_trino):
     )
 
 
-def test_int_types(trino_connection):
+def test_boolean_types(trino_connection):
     SqlTest(trino_connection) \
         .add_field(sql="false", python=False) \
+        .add_field(sql="true", python=True) \
         .add_field(sql="cast(null AS BOOLEAN)", python=None) \
+        .execute()
+
+
+def test_tinyint_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="cast(127 AS TINYINT)", python=127) \
         .add_field(sql="cast(-128 AS TINYINT)", python=-128) \
+        .add_field(sql="cast(42 AS TINYINT)", python=42) \
+        .add_field(sql="cast(-42 AS TINYINT)", python=-42) \
         .add_field(sql="cast(null AS TINYINT)", python=None) \
+        .execute()
+
+
+def test_smallint_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="cast(32767 AS SMALLINT)", python=32767) \
         .add_field(sql="cast(-32768 AS SMALLINT)", python=-32768) \
         .add_field(sql="cast(null AS SMALLINT)", python=None) \
+        .execute()
+
+
+def test_int_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="cast(2147483647 AS INTEGER)", python=2147483647) \
         .add_field(sql="cast(-2147483648 AS INTEGER)", python=-2147483648) \
         .add_field(sql="cast(null AS INTEGER)", python=None) \
@@ -34,25 +52,44 @@ def test_int_types(trino_connection):
         .execute()
 
 
-def test_float_types(trino_connection):
+def test_bigint_types(trino_connection):
+    SqlTest(trino_connection) \
+        .add_field(sql="cast(9223372036854775807 AS BIGINT)", python=9223372036854775807) \
+        .add_field(sql="cast(-9223372036854775808 AS BIGINT)", python=-9223372036854775808) \
+        .add_field(sql="cast(null AS BIGINT)", python=None) \
+        .execute()
+
+
+def test_real_types(trino_connection):
     SqlTest(trino_connection) \
         .add_field(sql="cast(3.4028235E38 AS REAL)", python=3.4028235e+38) \
         .add_field(sql="cast(1.4E-45 AS REAL)", python=1.4e-45) \
         .add_field(sql="cast('Infinity' AS REAL)", python=math.inf) \
         .add_field(sql="cast('-Infinity' AS REAL)", python=-math.inf) \
         .add_field(sql="cast('NaN' AS REAL)", python=math.nan) \
         .add_field(sql="cast(null AS REAL)", python=None) \
+        .execute()
+
+
+def test_double_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="cast(1.7976931348623157E308 AS DOUBLE)", python=1.7976931348623157e+308) \
         .add_field(sql="cast(4.9E-324 AS DOUBLE)", python=5e-324) \
         .add_field(sql="cast('Infinity' AS DOUBLE)", python=math.inf) \
         .add_field(sql="cast('-Infinity' AS DOUBLE)", python=-math.inf) \
         .add_field(sql="cast('NaN' AS DOUBLE)", python=math.nan) \
         .add_field(sql="cast(null AS DOUBLE)", python=None) \
+        .execute()
+
+
+def test_decimal_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="DECIMAL '10.3'", python=Decimal('10.3')) \
         .add_field(sql="cast(null AS DECIMAL)", python=None) \
         .add_field(sql="cast('0.123456789123456789' AS DECIMAL(18,18))", python=Decimal('0.123456789123456789')) \
         .add_field(sql="cast(null AS DECIMAL(18,18))", python=None) \
-        .add_field(sql="cast('10.3' AS DECIMAL(38,0))", python=10.0) \
+        .add_field(sql="cast('10.3' AS DECIMAL(38,1))", python=Decimal('10.3')) \
+        .add_field(sql="cast('0.123456789123456789' AS DECIMAL(18,2))", python=Decimal('0.12')) \
         .add_field(sql="cast(null AS DECIMAL(38,0))", python=None) \
         .execute()
 
@@ -67,6 +104,7 @@ def test_string_types(trino_connection):
         .add_field(sql="cast('ccc' AS CHAR)", python='c') \
         .add_field(sql="cast(null AS CHAR)", python=None) \
         .add_field(sql="cast('ddd' AS CHAR(1))", python='d') \
+        .add_field(sql="cast('😂' AS CHAR(1))", python='😂') \
         .add_field(sql="cast(null AS CHAR(1))", python=None) \
         .add_field(sql="X'65683F'", python='ZWg/') \
         .add_field(sql="cast(null AS VARBINARY)", python=None) \
@@ -81,6 +119,7 @@ def test_datetime_types(trino_connection):
 
     SqlTest(trino_connection) \
         .add_field(sql="DATE '2001-08-22'", python=datetime.date(2001, 8, 22)) \
+        .add_field(sql="DATE '02001-08-22'", python=datetime.date(2001, 8, 22)) \
         .add_field(sql="cast(null AS DATE)", python=None) \
         .add_field(sql="TIME '01:23:45.123'", python=datetime.time(1, 23, 45, 123000)) \
         .add_field(sql="cast(null AS TIME)", python=None) \
@@ -149,22 +188,53 @@ def test_datetime_types(trino_connection):
         .execute()
 
 
-def test_misc_types(trino_connection):
+def test_interval_types(trino_connection):
     SqlTest(trino_connection) \
         .add_field(sql="INTERVAL '3' MONTH", python='0-3') \
         .add_field(sql="cast(null AS INTERVAL YEAR TO MONTH)", python=None) \
         .add_field(sql="INTERVAL '2' DAY", python='2 00:00:00.000') \
+        .add_field(sql="INTERVAL '-2' DAY", python='-2 00:00:00.000') \
         .add_field(sql="cast(null AS INTERVAL DAY TO SECOND)", python=None) \
+        .execute()
+
+
+def test_array_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="ARRAY['a', 'b', null]", python=['a', 'b', None]) \
         .add_field(sql="cast(null AS ARRAY(VARCHAR))", python=None) \
+        .execute()
+
+
+def test_map_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="MAP(ARRAY['a', 'b'], ARRAY[1, null])", python={'a': 1, 'b': None}) \
         .add_field(sql="cast(null AS MAP(VARCHAR, INTEGER))", python=None) \
+        .execute()
+
+
+def test_row_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="cast(ROW(1, 2e0) AS ROW(x BIGINT, y DOUBLE))", python=(1, 2.0)) \
         .add_field(sql="cast(null AS ROW(x BIGINT, y DOUBLE))", python=None) \
+        .execute()
+
+
+def test_ipaddress_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="IPADDRESS '2001:db8::1'", python='2001:db8::1') \
         .add_field(sql="cast(null AS IPADDRESS)", python=None) \
+        .execute()
+
+
+def test_uuid_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="UUID '12151fd2-7586-11e9-8f9e-2a86e4085a59'", python='12151fd2-7586-11e9-8f9e-2a86e4085a59') \
         .add_field(sql="cast(null AS UUID)", python=None) \
+        .execute()
+
+
+def test_digest_types(trino_connection):
+    SqlTest(trino_connection) \
         .add_field(sql="approx_set(1)", python='AgwBAIADRAA=') \
         .add_field(sql="cast(null AS HyperLogLog)", python=None) \
         .add_field(sql="cast(approx_set(1) AS P4HyperLogLog)", python='AwwAAAAg' + 'A' * 2730 + '==') \