Fix types

python-pinot-dbapi · Apr 23, 2024 · f81e5c4 · f81e5c4
1 parent 11aa599
commit f81e5c4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 15 deletions.
diff --git a/pinotdb/sqlalchemy.py b/pinotdb/sqlalchemy.py
@@ -58,7 +58,7 @@ def visit_REAL(self, type_, **kwargs):
         return "DOUBLE"
 
     def visit_NUMERIC(self, type_, **kwargs):
-        return "LONG"
+        return "NUMERIC"
 
     visit_DECIMAL = visit_NUMERIC
     visit_INTEGER = visit_NUMERIC
@@ -69,7 +69,7 @@ def visit_NUMERIC(self, type_, **kwargs):
     visit_DATE = visit_NUMERIC
 
     def visit_CHAR(self, type_, **kwargs):
-        return "STRING"
+        return "VARCHAR"
 
     visit_NCHAR = visit_CHAR
     visit_VARCHAR = visit_CHAR

diff --git a/tests/unit/test_sqlalchemy.py b/tests/unit/test_sqlalchemy.py
@@ -357,44 +357,44 @@ def test_compiles_real(self):
         self.assertEqual(self.compiler.visit_REAL(None), 'DOUBLE')
 
     def test_compiles_numeric(self):
-        self.assertEqual(self.compiler.visit_NUMERIC(None), 'LONG')
+        self.assertEqual(self.compiler.visit_NUMERIC(None), 'NUMERIC')
 
     def test_compiles_decimal(self):
-        self.assertEqual(self.compiler.visit_DECIMAL(None), 'LONG')
+        self.assertEqual(self.compiler.visit_DECIMAL(None), 'NUMERIC')
 
     def test_compiles_integer(self):
-        self.assertEqual(self.compiler.visit_INTEGER(None), 'LONG')
+        self.assertEqual(self.compiler.visit_INTEGER(None), 'NUMERIC')
 
     def test_compiles_smallint(self):
-        self.assertEqual(self.compiler.visit_SMALLINT(None), 'LONG')
+        self.assertEqual(self.compiler.visit_SMALLINT(None), 'NUMERIC')
 
     def test_compiles_bigint(self):
-        self.assertEqual(self.compiler.visit_BIGINT(None), 'LONG')
+        self.assertEqual(self.compiler.visit_BIGINT(None), 'NUMERIC')
 
     # TODO: Check if this is correct (seems strange to have boolean as long).
     def test_compiles_boolean(self):
-        self.assertEqual(self.compiler.visit_BOOLEAN(None), 'LONG')
+        self.assertEqual(self.compiler.visit_BOOLEAN(None), 'NUMERIC')
 
     def test_compiles_timestamp(self):
-        self.assertEqual(self.compiler.visit_TIMESTAMP(None), 'LONG')
+        self.assertEqual(self.compiler.visit_TIMESTAMP(None), 'NUMERIC')
 
     def test_compiles_date(self):
-        self.assertEqual(self.compiler.visit_DATE(None), 'LONG')
+        self.assertEqual(self.compiler.visit_DATE(None), 'NUMERIC')
 
     def test_compiles_char(self):
-        self.assertEqual(self.compiler.visit_CHAR(None), 'STRING')
+        self.assertEqual(self.compiler.visit_CHAR(None), 'VARCHAR')
 
     def test_compiles_nchar(self):
-        self.assertEqual(self.compiler.visit_NCHAR(None), 'STRING')
+        self.assertEqual(self.compiler.visit_NCHAR(None), 'VARCHAR')
 
     def test_compiles_varchar(self):
-        self.assertEqual(self.compiler.visit_VARCHAR(None), 'STRING')
+        self.assertEqual(self.compiler.visit_VARCHAR(None), 'VARCHAR')
 
     def test_compiles_nvarchar(self):
-        self.assertEqual(self.compiler.visit_NVARCHAR(None), 'STRING')
+        self.assertEqual(self.compiler.visit_NVARCHAR(None), 'VARCHAR')
 
     def test_compiles_text(self):
-        self.assertEqual(self.compiler.visit_TEXT(None), 'STRING')
+        self.assertEqual(self.compiler.visit_TEXT(None), 'VARCHAR')
 
     def test_compiles_binary(self):
         self.assertEqual(self.compiler.visit_BINARY(None), 'BYTES')