fakesnow 0.4.1__tar.gz → 0.5.1__tar.gz

{fakesnow-0.4.1/fakesnow.egg-info → fakesnow-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.4.1
+Version: 0.5.1
 Summary: Fake Snowflake Connector for Python. Run Snowflake DB locally.
 License: MIT License
         
@@ -109,12 +109,13 @@ def _fakesnow_session() -> Iterator[None]:
 
 ## Implementation coverage
 
-- [x] multiple databases
-- [x] cursors
+- [x] cursors and standard SQL
 - [x] [get_result_batches()](https://docs.snowflake.com/en/user-guide/python-connector-api#get_result_batches)
-- [x] [write_pandas(..)](https://docs.snowflake.com/en/user-guide/python-connector-api#write_pandas)
-- [x] table comments
+- [x] information schema
+- [x] multiple databases
 - [x] [qmark binding](https://docs.snowflake.com/en/user-guide/python-connector-example#binding-data)
+- [x] table comments
+- [x] [write_pandas(..)](https://docs.snowflake.com/en/user-guide/python-connector-api#write_pandas)
 - [ ] [access control](https://docs.snowflake.com/en/user-guide/security-access-control-overview)
 - [ ] standalone/out of process api/support for faking non-python connectors
 - [ ] [stored procedures](https://docs.snowflake.com/en/sql-reference/stored-procedures)
@@ -122,8 +123,9 @@ def _fakesnow_session() -> Iterator[None]:
 Partial support
 
 - [x] date functions
-- [x] tags
+- [x] regex functions
 - [x] semi-structured data
+- [x] tags
 
 For more detail see [tests/test_fakes.py](tests/test_fakes.py)
 

{fakesnow-0.4.1 → fakesnow-0.5.1}/README.md

@@ -75,12 +75,13 @@ def _fakesnow_session() -> Iterator[None]:
 
 ## Implementation coverage
 
-- [x] multiple databases
-- [x] cursors
+- [x] cursors and standard SQL
 - [x] [get_result_batches()](https://docs.snowflake.com/en/user-guide/python-connector-api#get_result_batches)
-- [x] [write_pandas(..)](https://docs.snowflake.com/en/user-guide/python-connector-api#write_pandas)
-- [x] table comments
+- [x] information schema
+- [x] multiple databases
 - [x] [qmark binding](https://docs.snowflake.com/en/user-guide/python-connector-example#binding-data)
+- [x] table comments
+- [x] [write_pandas(..)](https://docs.snowflake.com/en/user-guide/python-connector-api#write_pandas)
 - [ ] [access control](https://docs.snowflake.com/en/user-guide/security-access-control-overview)
 - [ ] standalone/out of process api/support for faking non-python connectors
 - [ ] [stored procedures](https://docs.snowflake.com/en/sql-reference/stored-procedures)
@@ -88,8 +89,9 @@ def _fakesnow_session() -> Iterator[None]:
 Partial support
 
 - [x] date functions
-- [x] tags
+- [x] regex functions
 - [x] semi-structured data
+- [x] tags
 
 For more detail see [tests/test_fakes.py](tests/test_fakes.py)
 

{fakesnow-0.4.1 → fakesnow-0.5.1}/fakesnow/fakes.py

@@ -46,6 +46,7 @@ class FakeSnowflakeCursor:
         self._use_dict_result = use_dict_result
         self._last_sql = None
         self._last_params = None
+        self._sqlstate = None
 
     def __enter__(self) -> Self:
         return self
@@ -76,7 +77,7 @@ class FakeSnowflakeCursor:
         # use a cursor to avoid destroying an unfetched result on the main connection
         with self._duck_conn.cursor() as cur:
             assert self._conn.database, "Not implemented when database is None"
-            assert self._conn.schema, "Not implemented when database is None"
+            assert self._conn.schema, "Not implemented when schema is None"
 
             # match database and schema used on the main connection
             cur.execute(f"SET SCHEMA = '{self._conn.database}.{self._conn.schema}'")
@@ -91,6 +92,20 @@ class FakeSnowflakeCursor:
         params: Sequence[Any] | dict[Any, Any] | None = None,
         *args: Any,
         **kwargs: Any,
+    ) -> FakeSnowflakeCursor:
+        try:
+            self._sqlstate = None
+            return self._execute(command, params, *args, **kwargs)
+        except snowflake.connector.errors.ProgrammingError as e:
+            self._sqlstate = e.sqlstate
+            raise e
+
+    def _execute(
+        self,
+        command: str | exp.Expression,
+        params: Sequence[Any] | dict[Any, Any] | None = None,
+        *args: Any,
+        **kwargs: Any,
     ) -> FakeSnowflakeCursor:
         self._arrow_table = None
 
@@ -108,13 +123,13 @@ class FakeSnowflakeCursor:
                 msg=f"Cannot perform {cmd}. This session does not have a current database. Call 'USE DATABASE', or use a qualified name.",  # noqa: E501
                 errno=90105,
                 sqlstate="22000",
-            ) from None
+            )
         elif no_schema and not self._conn.schema_set:
             raise snowflake.connector.errors.ProgrammingError(
                 msg=f"Cannot perform {cmd}. This session does not have a current schema. Call 'USE SCHEMA', or use a qualified name.",  # noqa: E501
                 errno=90106,
                 sqlstate="22000",
-            ) from None
+            )
 
         transformed = (
             expression.transform(transforms.upper_case_unquoted_identifiers)
@@ -247,13 +262,21 @@ class FakeSnowflakeCursor:
         # TODO: return number of rows updated/inserted (using returning)
         return None
 
+    @property
+    def sfqid(self) -> str | None:
+        return "fakesnow"
+
+    @property
+    def sqlstate(self) -> str | None:
+        return self._sqlstate
+
     @staticmethod
     def _describe_as_result_metadata(describe_results: list) -> list[ResultMetadata]:
         # fmt: off
         def as_result_metadata(column_name: str, column_type: str, _: str) -> ResultMetadata:
             # see https://docs.snowflake.com/en/user-guide/python-connector-api.html#type-codes
             # and https://arrow.apache.org/docs/python/api/datatypes.html#type-checking
-            if column_type == "BIGINT":
+            if column_type in {"BIGINT", "INTEGER"}:
                 return ResultMetadata(
                     name=column_name, type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True               # noqa: E501
                 )
@@ -288,6 +311,14 @@ class FakeSnowflakeCursor:
                 return ResultMetadata(
                     name=column_name, type_code=8, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True                # noqa: E501
                 )
+            elif column_type == "BLOB":
+                return ResultMetadata(
+                    name=column_name, type_code=11, display_size=None, internal_size=8388608, precision=None, scale=None, is_nullable=True      # noqa: E501
+                )
+            elif column_type == "TIME":
+                return ResultMetadata(
+                    name=column_name, type_code=12, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True               # noqa: E501
+                )
             else:
                 # TODO handle more types
                 raise NotImplementedError(f"for column type {column_type}")

{fakesnow-0.4.1 → fakesnow-0.5.1}/fakesnow/info_schema.py

@@ -37,11 +37,18 @@ create view ${catalog}.information_schema.columns_snowflake AS
 select table_catalog, table_schema, table_name, column_name, ordinal_position, column_default, is_nullable,
 case when starts_with(data_type, 'DECIMAL') or data_type='BIGINT' then 'NUMBER'
      when data_type='VARCHAR' then 'TEXT'
+     when data_type='DOUBLE' then 'FLOAT'
+     when data_type='BLOB' then 'BINARY'
+     when data_type='TIMESTAMP' then 'TIMESTAMP_NTZ'
      else data_type end as data_type,
 ext_character_maximum_length as character_maximum_length, ext_character_octet_length as character_octet_length,
-case when data_type='BIGINT' then 38 else numeric_precision end as numeric_precision,
-case when data_type='BIGINT' then 10 else numeric_precision_radix end as numeric_precision_radix,
-numeric_scale,
+case when data_type='BIGINT' then 38
+     when data_type='DOUBLE' then NULL
+    else numeric_precision end as numeric_precision,
+case when data_type='BIGINT' then 10
+    when data_type='DOUBLE' then NULL
+    else numeric_precision_radix end as numeric_precision_radix,
+case when data_type='DOUBLE' then NULL else numeric_scale end as numeric_scale,
 collation_name, is_identity, identity_generation, identity_cycle
 from ${catalog}.information_schema.columns
 left join ${catalog}.information_schema.columns_ext ext

{fakesnow-0.4.1 → fakesnow-0.5.1}/fakesnow/transforms.py

@@ -181,10 +181,7 @@ def float_to_double(expression: exp.Expression) -> exp.Expression:
     """
 
     if isinstance(expression, exp.DataType) and expression.this == exp.DataType.Type.FLOAT:
-        # TODO don't copy!
-        new = expression.copy()
-        new.args["this"] = exp.DataType.Type.DOUBLE
-        return new
+        expression.args["this"] = exp.DataType.Type.DOUBLE
 
     return expression
 

{fakesnow-0.4.1 → fakesnow-0.5.1/fakesnow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.4.1
+Version: 0.5.1
 Summary: Fake Snowflake Connector for Python. Run Snowflake DB locally.
 License: MIT License
         
@@ -109,12 +109,13 @@ def _fakesnow_session() -> Iterator[None]:
 
 ## Implementation coverage
 
-- [x] multiple databases
-- [x] cursors
+- [x] cursors and standard SQL
 - [x] [get_result_batches()](https://docs.snowflake.com/en/user-guide/python-connector-api#get_result_batches)
-- [x] [write_pandas(..)](https://docs.snowflake.com/en/user-guide/python-connector-api#write_pandas)
-- [x] table comments
+- [x] information schema
+- [x] multiple databases
 - [x] [qmark binding](https://docs.snowflake.com/en/user-guide/python-connector-example#binding-data)
+- [x] table comments
+- [x] [write_pandas(..)](https://docs.snowflake.com/en/user-guide/python-connector-api#write_pandas)
 - [ ] [access control](https://docs.snowflake.com/en/user-guide/security-access-control-overview)
 - [ ] standalone/out of process api/support for faking non-python connectors
 - [ ] [stored procedures](https://docs.snowflake.com/en/sql-reference/stored-procedures)
@@ -122,8 +123,9 @@ def _fakesnow_session() -> Iterator[None]:
 Partial support
 
 - [x] date functions
-- [x] tags
+- [x] regex functions
 - [x] semi-structured data
+- [x] tags
 
 For more detail see [tests/test_fakes.py](tests/test_fakes.py)
 

{fakesnow-0.4.1 → fakesnow-0.5.1}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "fakesnow"
 description = "Fake Snowflake Connector for Python. Run Snowflake DB locally."
-version = "0.4.1"
+version = "0.5.1"
 readme = "README.md"
 license = { file = "LICENSE" }
 classifiers = ["License :: OSI Approved :: MIT License"]

{fakesnow-0.4.1 → fakesnow-0.5.1}/tests/test_fakes.py

@@ -1,3 +1,5 @@
+# ruff: noqa: E501
+
 import datetime
 import json
 
@@ -8,6 +10,7 @@ import snowflake.connector
 import snowflake.connector.cursor
 import snowflake.connector.pandas_tools
 from pandas.testing import assert_frame_equal
+from snowflake.connector.cursor import ResultMetadata
 
 
 def test_alter_table(cur: snowflake.connector.cursor.SnowflakeCursor):
@@ -62,7 +65,7 @@ def test_connect_without_database(_fakesnow_no_auto_create: None):
             cur.execute("SELECT * FROM jaffles.customers")
 
         assert (
-            "090105 (22000): Cannot perform SELECT. This session does not have a current database. Call 'USE DATABASE', or use a qualified name."  # noqa: E501
+            "090105 (22000): Cannot perform SELECT. This session does not have a current database. Call 'USE DATABASE', or use a qualified name."
             in str(excinfo.value)
         )
 
@@ -70,7 +73,7 @@ def test_connect_without_database(_fakesnow_no_auto_create: None):
             cur.execute("create schema jaffles")
 
         assert (
-            "090105 (22000): Cannot perform CREATE SCHEMA. This session does not have a current database. Call 'USE DATABASE', or use a qualified name."  # noqa: E501
+            "090105 (22000): Cannot perform CREATE SCHEMA. This session does not have a current database. Call 'USE DATABASE', or use a qualified name."
             in str(excinfo.value)
         )
 
@@ -86,7 +89,7 @@ def test_connect_without_database(_fakesnow_no_auto_create: None):
             cur.execute("create table customers (ID int, FIRST_NAME varchar, LAST_NAME varchar)")
 
         assert (
-            "090105 (22000): Cannot perform CREATE TABLE. This session does not have a current database. Call 'USE DATABASE', or use a qualified name."  # noqa: E501
+            "090105 (22000): Cannot perform CREATE TABLE. This session does not have a current database. Call 'USE DATABASE', or use a qualified name."
             in str(excinfo.value)
         )
 
@@ -112,7 +115,7 @@ def test_connect_without_schema(_fakesnow: None):
             cur.execute("create table customers (ID int, FIRST_NAME varchar, LAST_NAME varchar)")
 
         assert (
-            "090106 (22000): Cannot perform CREATE TABLE. This session does not have a current schema. Call 'USE SCHEMA', or use a qualified name."  # noqa: E501
+            "090106 (22000): Cannot perform CREATE TABLE. This session does not have a current schema. Call 'USE SCHEMA', or use a qualified name."
             in str(excinfo.value)
         )
 
@@ -148,7 +151,7 @@ def test_connect_with_non_existent_db_or_schema(_fakesnow_no_auto_create: None):
             cur.execute("create table foobar (i int)")
 
         assert (
-            "090105 (22000): Cannot perform CREATE TABLE. This session does not have a current database. Call 'USE DATABASE', or use a qualified name."  # noqa: E501
+            "090105 (22000): Cannot perform CREATE TABLE. This session does not have a current database. Call 'USE DATABASE', or use a qualified name."
             in str(excinfo.value)
         )
 
@@ -164,7 +167,7 @@ def test_connect_with_non_existent_db_or_schema(_fakesnow_no_auto_create: None):
             cur.execute("create table foobar (i int)")
 
         assert (
-            "090106 (22000): Cannot perform CREATE TABLE. This session does not have a current schema. Call 'USE SCHEMA', or use a qualified name."  # noqa: E501
+            "090106 (22000): Cannot perform CREATE TABLE. This session does not have a current schema. Call 'USE SCHEMA', or use a qualified name."
             in str(excinfo.value)
         )
 
@@ -184,88 +187,64 @@ def test_current_database_schema(conn: snowflake.connector.SnowflakeConnection):
 def test_describe(cur: snowflake.connector.cursor.SnowflakeCursor):
     cur.execute(
         """
-        create table customers (
-            ID int, CNAME varchar, AMOUNT decimal(10,2), PCT real, ACTIVE boolean,
-            UPDATE_AT timestamp, UPDATE_AT_NTZ timestamp_ntz(9), INSERTIONDATE DATE
+        create or replace table example (
+            XBOOLEAN BOOLEAN, XDOUBLE DOUBLE, XFLOAT FLOAT,
+            XNUMBER82 NUMBER(8,2), XNUMBER NUMBER, XDECIMAL DECIMAL, XNUMERIC NUMERIC,
+            XINT INT, XINTEGER INTEGER, XBIGINT BIGINT, XSMALLINT SMALLINT, XTINYINT TINYINT, XBYTEINT BYTEINT,
+            XVARCHAR20 VARCHAR(20), XVARCHAR VARCHAR, XTEXT TEXT,
+            XTIMESTAMP TIMESTAMP, XTIMESTAMP_NTZ9 TIMESTAMP_NTZ(9), XDATE DATE, XTIME TIME,
+            XBINARY BINARY
         )
         """
     )
     # fmt: off
     expected_metadata = [
-        snowflake.connector.cursor.ResultMetadata(
-            name="ID", type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True                  # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name="CNAME", type_code=2, display_size=None, internal_size=16777216, precision=None, scale=None, is_nullable=True,     # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name="AMOUNT", type_code=0, display_size=None, internal_size=None, precision=10, scale=2, is_nullable=True,             # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name="PCT", type_code=1, display_size=None, internal_size=None, precision=None, scale=None, is_nullable=True,           # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name="ACTIVE", type_code=13, display_size=None, internal_size=None, precision=None, scale=None, is_nullable=True,       # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name='UPDATE_AT', type_code=8, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True            # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name='UPDATE_AT_NTZ', type_code=8, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True        # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name='INSERTIONDATE', type_code=3, display_size=None, internal_size=None, precision=None, scale=None, is_nullable=True  # type: ignore # noqa: E501
-        ),
+        ResultMetadata(name='XBOOLEAN', type_code=13, display_size=None, internal_size=None, precision=None, scale=None, is_nullable=True),
+        ResultMetadata(name='XDOUBLE', type_code=1, display_size=None, internal_size=None, precision=None, scale=None, is_nullable=True),
+        ResultMetadata(name='XFLOAT', type_code=1, display_size=None, internal_size=None, precision=None, scale=None, is_nullable=True),
+        ResultMetadata(name='XNUMBER82', type_code=0, display_size=None, internal_size=None, precision=8, scale=2, is_nullable=True),
+        ResultMetadata(name='XNUMBER', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True),
+        ResultMetadata(name='XDECIMAL', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True),
+        ResultMetadata(name='XNUMERIC', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True),
+        ResultMetadata(name='XINT', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True),
+        ResultMetadata(name='XINTEGER', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True),
+        ResultMetadata(name='XBIGINT', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True),
+        ResultMetadata(name='XSMALLINT', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True),
+        ResultMetadata(name='XTINYINT', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True),
+        ResultMetadata(name='XBYTEINT', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True),
+        # TODO: store actual size
+        ResultMetadata(name='XVARCHAR20', type_code=2, display_size=None, internal_size=16777216, precision=None, scale=None, is_nullable=True),
+        ResultMetadata(name='XVARCHAR', type_code=2, display_size=None, internal_size=16777216, precision=None, scale=None, is_nullable=True),
+        ResultMetadata(name='XTEXT', type_code=2, display_size=None, internal_size=16777216, precision=None, scale=None, is_nullable=True),
+        ResultMetadata(name='XTIMESTAMP', type_code=8, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True),
+        ResultMetadata(name='XTIMESTAMP_NTZ9', type_code=8, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True),
+        ResultMetadata(name='XDATE', type_code=3, display_size=None, internal_size=None, precision=None, scale=None, is_nullable=True),
+        ResultMetadata(name='XTIME', type_code=12, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True),
+        ResultMetadata(name='XBINARY', type_code=11, display_size=None, internal_size=8388608, precision=None, scale=None, is_nullable=True)
     ]
     # fmt: on
 
-    assert cur.describe("select * from customers") == expected_metadata
+    assert cur.describe("select * from example") == expected_metadata
+    cur.execute("select * from example")
+    assert cur.description == expected_metadata
 
-    cur.execute("select * from customers")
+    # test with params
+    assert cur.describe("select * from example where XNUMBER = ?", (1,)) == expected_metadata
+    cur.execute("select * from example where XNUMBER = ?", (1,))
     assert cur.description == expected_metadata
 
 
-def test_describe_with_params(cur: snowflake.connector.cursor.SnowflakeCursor):
-    cur.execute(
-        """
-        create table customers (
-            ID int, CNAME varchar, AMOUNT decimal(10,2), PCT real, ACTIVE boolean,
-            UPDATE_AT timestamp, UPDATE_AT_NTZ timestamp_ntz(9), INSERTIONDATE DATE
-        )
-        """
-    )
+def test_describe_info_schema(cur: snowflake.connector.cursor.SnowflakeCursor):
+    # tests we can handle the column types returned from the info schema, which are created by duckdb
+    # and so don't go through our transforms
+    cur.execute("select column_name, ordinal_position from information_schema.columns")
     # fmt: off
     expected_metadata = [
-        snowflake.connector.cursor.ResultMetadata(
-            name="ID", type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True                  # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name="CNAME", type_code=2, display_size=None, internal_size=16777216, precision=None, scale=None, is_nullable=True,     # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name="AMOUNT", type_code=0, display_size=None, internal_size=None, precision=10, scale=2, is_nullable=True,             # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name="PCT", type_code=1, display_size=None, internal_size=None, precision=None, scale=None, is_nullable=True,           # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name="ACTIVE", type_code=13, display_size=None, internal_size=None, precision=None, scale=None, is_nullable=True,       # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name='UPDATE_AT', type_code=8, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True            # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name='UPDATE_AT_NTZ', type_code=8, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True        # type: ignore # noqa: E501
-        ),
-        snowflake.connector.cursor.ResultMetadata(
-            name='INSERTIONDATE', type_code=3, display_size=None, internal_size=None, precision=None, scale=None, is_nullable=True  # type: ignore # noqa: E501
-        ),
+        ResultMetadata(name='column_name', type_code=2, display_size=None, internal_size=16777216, precision=None, scale=None, is_nullable=True),
+        ResultMetadata(name='ordinal_position', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True)
     ]
     # fmt: on
 
-    assert cur.describe("select * from customers where id = ?", (1,)) == expected_metadata
-
-    cur.execute("select * from customers where id = ?", (1,))
     assert cur.description == expected_metadata
 
 
@@ -390,8 +369,8 @@ def test_information_schema_columns_numeric(cur: snowflake.connector.cursor.Snow
     # see https://docs.snowflake.com/en/sql-reference/data-types-numeric
     cur.execute(
         """
-        create table example (
-            XNUMBER82 NUMBER(8,2), XNUMBER NUMBER, XDECIMAL DECIMAL, XNUMERIC NUMERIC,
+        create or replace table example (
+            XBOOLEAN BOOLEAN, XDOUBLE DOUBLE, XFLOAT FLOAT, XNUMBER82 NUMBER(8,2), XNUMBER NUMBER, XDECIMAL DECIMAL, XNUMERIC NUMERIC,
             XINT INT, XINTEGER INTEGER, XBIGINT BIGINT, XSMALLINT SMALLINT, XTINYINT TINYINT, XBYTEINT BYTEINT
         )
         """
@@ -405,6 +384,9 @@ def test_information_schema_columns_numeric(cur: snowflake.connector.cursor.Snow
     )
 
     assert cur.fetchall() == [
+        ("XBOOLEAN", "BOOLEAN", None, None, None),
+        ("XDOUBLE", "FLOAT", None, None, None),
+        ("XFLOAT", "FLOAT", None, None, None),
         ("XNUMBER82", "NUMBER", 8, 10, 2),
         ("XNUMBER", "NUMBER", 38, 10, 0),
         ("XDECIMAL", "NUMBER", 38, 10, 0),
@@ -418,11 +400,38 @@ def test_information_schema_columns_numeric(cur: snowflake.connector.cursor.Snow
     ]
 
 
+def test_information_schema_columns_other(cur: snowflake.connector.cursor.SnowflakeCursor):
+    # see https://docs.snowflake.com/en/sql-reference/data-types-datetime
+    cur.execute(
+        """
+        create or replace table example (
+            XTIMESTAMP TIMESTAMP, XTIMESTAMP_NTZ9 TIMESTAMP_NTZ(9), XDATE DATE, XTIME TIME,
+            XBINARY BINARY
+        )
+        """
+    )
+
+    cur.execute(
+        """
+        select column_name,data_type
+        from information_schema.columns where table_name = 'EXAMPLE' order by ordinal_position
+        """
+    )
+
+    assert cur.fetchall() == [
+        ("XTIMESTAMP", "TIMESTAMP_NTZ"),
+        ("XTIMESTAMP_NTZ9", "TIMESTAMP_NTZ"),
+        ("XDATE", "DATE"),
+        ("XTIME", "TIME"),
+        ("XBINARY", "BINARY"),
+    ]
+
+
 def test_information_schema_columns_text(cur: snowflake.connector.cursor.SnowflakeCursor):
     # see https://docs.snowflake.com/en/sql-reference/data-types-text
     cur.execute(
         """
-        create table example (
+        create or replace table example (
             XVARCHAR20 VARCHAR(20), XVARCHAR VARCHAR, XTEXT TEXT
         )
         """
@@ -507,7 +516,7 @@ def test_semi_structured_types(cur: snowflake.connector.cursor.SnowflakeCursor):
         """insert into semis(emails, name, notes) SELECT [1, 2], parse_json('{"k": "v1"}'), parse_json('["foo"]')"""
     )
     cur.execute(
-        """insert into semis(emails, name, notes) VALUES ([3,4], parse_json('{"k": "v2"}'), parse_json('{"b": "ar"}'))"""  # noqa: E501
+        """insert into semis(emails, name, notes) VALUES ([3,4], parse_json('{"k": "v2"}'), parse_json('{"b": "ar"}'))"""
     )
 
     # results are returned as strings, because the underlying type is JSON (duckdb) / VARIANT (snowflake)
@@ -522,6 +531,21 @@ def test_semi_structured_types(cur: snowflake.connector.cursor.SnowflakeCursor):
     assert cur.fetchall() == [('"foo"',), (None,)]
 
 
+def test_sqlstate(cur: snowflake.connector.cursor.SnowflakeCursor):
+    cur.execute("select 'hello world'")
+    # sqlstate is None on success
+    assert cur.sqlstate is None
+
+    with pytest.raises(snowflake.connector.errors.ProgrammingError) as _:
+        cur.execute("select * from this_table_does_not_exist")
+
+    assert cur.sqlstate == "42S02"
+
+
+def test_sfqid(cur: snowflake.connector.cursor.SnowflakeCursor):
+    assert cur.sfqid == "fakesnow"
+
+
 def test_table_comments(cur: snowflake.connector.cursor.SnowflakeCursor):
     def read_comment() -> str:
         cur.execute(
@@ -626,7 +650,7 @@ def test_use_invalid_schema(_fakesnow: None):
             cur.execute("create table foobar (i int)")
 
         assert (
-            "090106 (22000): Cannot perform CREATE TABLE. This session does not have a current schema. Call 'USE SCHEMA', or use a qualified name."  # noqa: E501
+            "090106 (22000): Cannot perform CREATE TABLE. This session does not have a current schema. Call 'USE SCHEMA', or use a qualified name."
             in str(excinfo.value)
         )