fakesnow 0.9.9__tar.gz → 0.9.11__tar.gz

{fakesnow-0.9.9 → fakesnow-0.9.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.9.9
+Version: 0.9.11
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
 License:                                  Apache License
                                    Version 2.0, January 2004
@@ -220,7 +220,7 @@ Requires-Dist: pandas-stubs; extra == "dev"
 Requires-Dist: snowflake-connector-python[pandas,secure-local-storage]; extra == "dev"
 Requires-Dist: pre-commit~=3.4; extra == "dev"
 Requires-Dist: pytest~=8.0; extra == "dev"
-Requires-Dist: ruff~=0.3.2; extra == "dev"
+Requires-Dist: ruff~=0.4.2; extra == "dev"
 Requires-Dist: twine~=5.0; extra == "dev"
 Requires-Dist: snowflake-sqlalchemy~=1.5.0; extra == "dev"
 Provides-Extra: notebook
@@ -357,7 +357,7 @@ For more detail see [tests/test_fakes.py](tests/test_fakes.py)
 ## Caveats
 
 - The order of rows is non deterministic and may not match Snowflake unless ORDER BY is fully specified.
-- A more liberal Snowflake SQL dialect than a real Snowflake instance is supported, ie: some queries might pass using fakesnow that a real Snowflake instance would reject.
+- A more liberal Snowflake SQL dialect than used by a real Snowflake instance is supported, ie: some queries might pass using fakesnow that a real Snowflake instance would reject.
 
 ## Contributing
 

{fakesnow-0.9.9 → fakesnow-0.9.11}/README.md

@@ -127,7 +127,7 @@ For more detail see [tests/test_fakes.py](tests/test_fakes.py)
 ## Caveats
 
 - The order of rows is non deterministic and may not match Snowflake unless ORDER BY is fully specified.
-- A more liberal Snowflake SQL dialect than a real Snowflake instance is supported, ie: some queries might pass using fakesnow that a real Snowflake instance would reject.
+- A more liberal Snowflake SQL dialect than used by a real Snowflake instance is supported, ie: some queries might pass using fakesnow that a real Snowflake instance would reject.
 
 ## Contributing
 

{fakesnow-0.9.9 → fakesnow-0.9.11}/fakesnow/fakes.py

@@ -193,6 +193,7 @@ class FakeSnowflakeCursor:
             .transform(transforms.show_users)
             .transform(transforms.create_user)
             .transform(transforms.sha256)
+            .transform(transforms.create_clone)
         )
 
     def _execute(
@@ -629,7 +630,12 @@ class FakeSnowflakeConnection:
             df[col] = df[col].apply(lambda x: json.dumps(x) if isinstance(x, (dict, list)) else x)
 
         escaped_cols = ",".join(f'"{col}"' for col in df.columns.to_list())
-        self._duck_conn.execute(f"INSERT INTO {table_name}({escaped_cols}) SELECT * FROM df")
+        name = table_name
+        if schema:
+            table_name = f"{schema}.{table_name}"
+        if database:
+            name = f"{database}.{table_name}"
+        self._duck_conn.execute(f"INSERT INTO {name}({escaped_cols}) SELECT * FROM df")
 
         return self._duck_conn.fetchall()[0][0]
 

{fakesnow-0.9.9 → fakesnow-0.9.11}/fakesnow/transforms.py

@@ -55,6 +55,27 @@ def array_agg_within_group(expression: exp.Expression) -> exp.Expression:
     return expression
 
 
+def create_clone(expression: exp.Expression) -> exp.Expression:
+    """Transform create table clone to create table as select."""
+
+    if (
+        isinstance(expression, exp.Create)
+        and str(expression.args.get("kind")).upper() == "TABLE"
+        and (clone := expression.find(exp.Clone))
+    ):
+        return exp.Create(
+            this=expression.this,
+            kind="TABLE",
+            expression=exp.Select(
+                expressions=[
+                    exp.Star(),
+                ],
+                **{"from": exp.From(this=clone.this)},
+            ),
+        )
+    return expression
+
+
 # TODO: move this into a Dialect as a transpilation
 def create_database(expression: exp.Expression, db_path: Path | None = None) -> exp.Expression:
     """Transform create database to attach database.

{fakesnow-0.9.9 → fakesnow-0.9.11}/fakesnow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.9.9
+Version: 0.9.11
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
 License:                                  Apache License
                                    Version 2.0, January 2004
@@ -220,7 +220,7 @@ Requires-Dist: pandas-stubs; extra == "dev"
 Requires-Dist: snowflake-connector-python[pandas,secure-local-storage]; extra == "dev"
 Requires-Dist: pre-commit~=3.4; extra == "dev"
 Requires-Dist: pytest~=8.0; extra == "dev"
-Requires-Dist: ruff~=0.3.2; extra == "dev"
+Requires-Dist: ruff~=0.4.2; extra == "dev"
 Requires-Dist: twine~=5.0; extra == "dev"
 Requires-Dist: snowflake-sqlalchemy~=1.5.0; extra == "dev"
 Provides-Extra: notebook
@@ -357,7 +357,7 @@ For more detail see [tests/test_fakes.py](tests/test_fakes.py)
 ## Caveats
 
 - The order of rows is non deterministic and may not match Snowflake unless ORDER BY is fully specified.
-- A more liberal Snowflake SQL dialect than a real Snowflake instance is supported, ie: some queries might pass using fakesnow that a real Snowflake instance would reject.
+- A more liberal Snowflake SQL dialect than used by a real Snowflake instance is supported, ie: some queries might pass using fakesnow that a real Snowflake instance would reject.
 
 ## Contributing
 

{fakesnow-0.9.9 → fakesnow-0.9.11}/fakesnow.egg-info/requires.txt

@@ -9,7 +9,7 @@ pandas-stubs
 snowflake-connector-python[pandas,secure-local-storage]
 pre-commit~=3.4
 pytest~=8.0
-ruff~=0.3.2
+ruff~=0.4.2
 twine~=5.0
 snowflake-sqlalchemy~=1.5.0
 

{fakesnow-0.9.9 → fakesnow-0.9.11}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "fakesnow"
 description = "Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally."
-version = "0.9.9"
+version = "0.9.11"
 readme = "README.md"
 license = { file = "LICENSE" }
 classifiers = ["License :: OSI Approved :: MIT License"]
@@ -29,7 +29,7 @@ dev = [
     "snowflake-connector-python[pandas, secure-local-storage]",
     "pre-commit~=3.4",
     "pytest~=8.0",
-    "ruff~=0.3.2",
+    "ruff~=0.4.2",
     "twine~=5.0",
     "snowflake-sqlalchemy~=1.5.0",
 ]

{fakesnow-0.9.9 → fakesnow-0.9.11}/tests/test_fakes.py

@@ -110,6 +110,16 @@ def test_binding_qmark(_fakesnow: None):
         cur.execute("select * from customers where id = ?", (1,))
 
 
+def test_clone(cur: snowflake.connector.cursor.SnowflakeCursor):
+    cur.execute("create table customers (ID int, FIRST_NAME varchar, ACTIVE boolean)")
+    cur.execute("insert into customers values (1, 'Jenny', True)")
+
+    cur.execute("create table customers2 clone db1.schema1.customers")
+    cur.execute("select * from customers2")
+    # TODO check tags are copied too
+    assert cur.fetchall() == [(1, "Jenny", True)]
+
+
 def test_close_conn(conn: snowflake.connector.SnowflakeConnection, cur: snowflake.connector.cursor.SnowflakeCursor):
     conn.close()
     with pytest.raises(snowflake.connector.errors.DatabaseError) as excinfo:
@@ -1601,6 +1611,26 @@ def test_write_pandas_dict_different_keys(conn: snowflake.connector.SnowflakeCon
         assert indent(cur.fetchall()) == [('{\n  "k": "v1"\n}',), ('{\n  "k2": [\n    "v\'2",\n    "v\\"3"\n  ]\n}',)]
 
 
+def test_write_pandas_db_schema(conn: snowflake.connector.SnowflakeConnection):
+    with conn.cursor() as cur:
+        cur.execute("create database db2")
+        cur.execute("create schema db2.schema2")
+        cur.execute("create or replace table db2.schema2.customers (ID int, FIRST_NAME varchar, LAST_NAME varchar)")
+
+        df = pd.DataFrame.from_records(
+            [
+                {"ID": 1, "FIRST_NAME": "Jenny"},
+                {"ID": 2, "FIRST_NAME": "Jasper"},
+            ]
+        )
+        snowflake.connector.pandas_tools.write_pandas(conn, df, "CUSTOMERS", "DB2", "SCHEMA2")
+
+        cur.execute("select id, first_name, last_name from db2.schema2.customers")
+
+        # columns not in dataframe will receive their default value
+        assert cur.fetchall() == [(1, "Jenny", None), (2, "Jasper", None)]
+
+
 def indent(rows: Sequence[tuple] | Sequence[dict]) -> list[tuple]:
     # indent duckdb json strings tuple values to match snowflake json strings
     assert isinstance(rows[0], tuple)

{fakesnow-0.9.9 → fakesnow-0.9.11}/tests/test_transforms.py

@@ -9,6 +9,7 @@ from fakesnow.transforms import (
     _get_to_number_args,
     array_agg_within_group,
     array_size,
+    create_clone,
     create_database,
     dateadd_date_cast,
     dateadd_string_literal_timestamp_cast,
@@ -84,6 +85,13 @@ def test_array_agg_within_group() -> None:
     )
 
 
+def test_create_clone() -> None:
+    assert (
+        sqlglot.parse_one("create table customers2 clone db1.schema1.customer").transform(create_clone).sql()
+        == "CREATE TABLE customers2 AS SELECT * FROM db1.schema1.customer"
+    )
+
+
 def test_create_database() -> None:
     e = sqlglot.parse_one("create database foobar").transform(create_database)
     assert e.sql() == "ATTACH DATABASE ':memory:' AS foobar"