sql-testing-library 0.15.0__tar.gz → 0.17.0__tar.gz

{sql_testing_library-0.15.0 → sql_testing_library-0.17.0}/CHANGELOG.md

@@ -5,6 +5,28 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## 0.17.0 (2025-11-20)
+
+### Feat
+
+- **athena**: add option to specify workgroup (#127)
+- add CI code quality checks for all PRs (#129)
+
+### Fix
+
+- correct workflow condition to allow push events after merge (#130)
+- skip integration tests for fork PRs due to secret restrictions (#128)
+
+## 0.16.0 (2025-08-20)
+
+### Feat
+
+- upgrade mocksmith to 6.0.1 (#122)
+
+### Fix
+
+- upgrade snowflake-connector-python to >=3.13.1 for critical security vulnerabilities (#123)
+
 ## 0.15.0 (2025-07-27)
 
 ### Feat

{sql_testing_library-0.15.0 → sql_testing_library-0.17.0}/PKG-INFO

@@ -1,8 +1,9 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: sql-testing-library
-Version: 0.15.0
+Version: 0.17.0
 Summary: A powerful Python framework for unit testing SQL queries across BigQuery, Snowflake, Redshift, Athena, Trino, and DuckDB with mock data
 License: MIT
+License-File: LICENSE
 Keywords: sql,testing,unit-testing,mock-data,database-testing,bigquery,snowflake,redshift,athena,trino,duckdb,data-engineering,etl-testing,sql-validation,query-testing
 Author: Gurmeet Saran
 Author-email: gurmeetx@gmail.com
@@ -24,6 +25,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Topic :: Database
 Classifier: Topic :: Database :: Database Engines/Servers
@@ -40,7 +42,7 @@ Provides-Extra: redshift
 Provides-Extra: snowflake
 Provides-Extra: trino
 Requires-Dist: db-dtypes (>=1.4.3,<2.0.0)
-Requires-Dist: docutils (>=0.21.2,<0.22.0)
+Requires-Dist: docutils (>=0.21.2,<0.23.0)
 Requires-Dist: pandas (>=1.0.0)
 Requires-Dist: pydantic (>=2.0.0)
 Requires-Dist: sqlglot (>=26.0.0)
@@ -66,6 +68,7 @@ A powerful Python framework for unit testing SQL queries with mock data injectio
 [![Redshift Integration](https://github.com/gurmeetsaran/sqltesting/actions/workflows/redshift-integration.yml/badge.svg)](https://github.com/gurmeetsaran/sqltesting/actions/workflows/redshift-integration.yml)
 [![Trino Integration](https://github.com/gurmeetsaran/sqltesting/actions/workflows/trino-integration.yml/badge.svg)](https://github.com/gurmeetsaran/sqltesting/actions/workflows/trino-integration.yml)
 [![Snowflake Integration](https://github.com/gurmeetsaran/sqltesting/actions/workflows/snowflake-integration.yml/badge.svg)](https://github.com/gurmeetsaran/sqltesting/actions/workflows/snowflake-integration.yml)
+[![DuckDB Integration](https://github.com/gurmeetsaran/sqltesting/actions/workflows/duckdb-integration.yml/badge.svg)](https://github.com/gurmeetsaran/sqltesting/actions/workflows/duckdb-integration.yml)
 [![GitHub license](https://img.shields.io/github/license/gurmeetsaran/sqltesting)](https://github.com/gurmeetsaran/sqltesting/blob/master/LICENSE)
 [![Pepy Total Downloads](https://img.shields.io/pepy/dt/sql-testing-library?label=PyPI%20Downloads)](https://pepy.tech/projects/sql-testing-library)
 [![codecov](https://codecov.io/gh/gurmeetsaran/sqltesting/branch/master/graph/badge.svg?token=CN3G5X5ZA5)](https://codecov.io/gh/gurmeetsaran/sqltesting)

{sql_testing_library-0.15.0 → sql_testing_library-0.17.0}/README.md

@@ -8,6 +8,7 @@ A powerful Python framework for unit testing SQL queries with mock data injectio
 [![Redshift Integration](https://github.com/gurmeetsaran/sqltesting/actions/workflows/redshift-integration.yml/badge.svg)](https://github.com/gurmeetsaran/sqltesting/actions/workflows/redshift-integration.yml)
 [![Trino Integration](https://github.com/gurmeetsaran/sqltesting/actions/workflows/trino-integration.yml/badge.svg)](https://github.com/gurmeetsaran/sqltesting/actions/workflows/trino-integration.yml)
 [![Snowflake Integration](https://github.com/gurmeetsaran/sqltesting/actions/workflows/snowflake-integration.yml/badge.svg)](https://github.com/gurmeetsaran/sqltesting/actions/workflows/snowflake-integration.yml)
+[![DuckDB Integration](https://github.com/gurmeetsaran/sqltesting/actions/workflows/duckdb-integration.yml/badge.svg)](https://github.com/gurmeetsaran/sqltesting/actions/workflows/duckdb-integration.yml)
 [![GitHub license](https://img.shields.io/github/license/gurmeetsaran/sqltesting)](https://github.com/gurmeetsaran/sqltesting/blob/master/LICENSE)
 [![Pepy Total Downloads](https://img.shields.io/pepy/dt/sql-testing-library?label=PyPI%20Downloads)](https://pepy.tech/projects/sql-testing-library)
 [![codecov](https://codecov.io/gh/gurmeetsaran/sqltesting/branch/master/graph/badge.svg?token=CN3G5X5ZA5)](https://codecov.io/gh/gurmeetsaran/sqltesting)

{sql_testing_library-0.15.0 → sql_testing_library-0.17.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "sql-testing-library"
-version = "0.15.0"
+version = "0.17.0"
 description = "A powerful Python framework for unit testing SQL queries across BigQuery, Snowflake, Redshift, Athena, Trino, and DuckDB with mock data"
 authors = ["Gurmeet Saran <gurmeetx@gmail.com>", "Kushal Thakkar <kushal.thakkar@gmail.com>"]
 maintainers = ["Gurmeet Saran <gurmeetx@gmail.com>", "Kushal Thakkar <kushal.thakkar@gmail.com>"]
@@ -74,7 +74,7 @@ sqlglot = ">=26.0.0"
 pydantic = ">=2.0.0"
 pandas = ">=1.0.0"
 typing-extensions = {version = ">=4.5.0", python = "<3.10"}
-docutils = ">=0.21.2,<0.22.0"
+docutils = ">=0.21.2,<0.23.0"
 db-dtypes = "^1.4.3"
 
 [tool.poetry.group.bigquery.dependencies]
@@ -90,7 +90,7 @@ psycopg2-binary = ">=2.9.0"
 trino = ">=0.322.0"
 
 [tool.poetry.group.snowflake.dependencies]
-snowflake-connector-python = ">=3.0.0"
+snowflake-connector-python = ">=3.13.1"
 
 [tool.poetry.group.duckdb.dependencies]
 duckdb = ">=0.8.0"
@@ -120,8 +120,8 @@ twine = ">=4.0.0"
 ruff = ">=0.0.287"
 pre-commit = ">=3.5.0"
 commitizen = ">=3.0.0"
-mocksmith = {version = ">=3.0.3", extras = ["mock", "pydantic"]}
 faker = ">=18.0.0"  # Explicit dependency to ensure it's installed in CI
+mocksmith = ">=6.0.1"
 
 [tool.poetry.plugins.pytest11]
 sql_testing = "sql_testing_library._pytest_plugin"

{sql_testing_library-0.15.0 → sql_testing_library-0.17.0}/src/sql_testing_library/_adapters/athena.py

@@ -44,6 +44,7 @@ class AthenaAdapter(PrestoBaseAdapter):
         database: str,
         s3_output_location: str,
         region: str = "us-west-2",
+        workgroup: Optional[str] = None,
         aws_access_key_id: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
     ) -> None:
@@ -58,6 +59,7 @@ class AthenaAdapter(PrestoBaseAdapter):
         self.database = database
         self.s3_output_location = s3_output_location
         self.region = region
+        self.workgroup = workgroup
 
         # Initialize Athena client
         if aws_access_key_id and aws_secret_access_key:
@@ -79,11 +81,18 @@ class AthenaAdapter(PrestoBaseAdapter):
         """Execute query and return results as DataFrame."""
         import pandas as pd
 
+        query_params: dict[str, Any] = {
+            "QueryExecutionContext": {"Database": self.database},
+            "ResultConfiguration": {"OutputLocation": self.s3_output_location},
+        }
+
+        if self.workgroup is not None:
+            query_params["WorkGroup"] = self.workgroup
+
         # Start query execution
         response = self.client.start_query_execution(
             QueryString=query,
-            QueryExecutionContext={"Database": self.database},
-            ResultConfiguration={"OutputLocation": self.s3_output_location},
+            **query_params,
         )
 
         query_execution_id = response["QueryExecutionId"]

{sql_testing_library-0.15.0 → sql_testing_library-0.17.0}/src/sql_testing_library/_pytest_plugin.py

@@ -86,6 +86,7 @@ class SQLTestDecorator:
             database = adapter_config.get("database")
             s3_output_location = adapter_config.get("s3_output_location")
             region = adapter_config.get("region", "us-west-2")
+            workgroup = adapter_config.get("workgroup")
             aws_access_key_id = adapter_config.get("aws_access_key_id")
             aws_secret_access_key = adapter_config.get("aws_secret_access_key")
 
@@ -98,6 +99,7 @@ class SQLTestDecorator:
                 database=database,
                 s3_output_location=s3_output_location,
                 region=region,
+                workgroup=workgroup,
                 aws_access_key_id=aws_access_key_id,
                 aws_secret_access_key=aws_secret_access_key,
             )

{sql_testing_library-0.15.0 → sql_testing_library-0.17.0}/src/sql_testing_library/_sql_utils.py

@@ -1,5 +1,6 @@
 """SQL utility functions for escaping and formatting values."""
 
+import inspect
 import json
 from dataclasses import is_dataclass
 from datetime import date, datetime
@@ -469,14 +470,16 @@ def format_sql_value(value: Any, column_type: Type, dialect: str = "standard") -
         else:
             return escape_sql_string(str(value))
 
-    # Handle numeric types
-    elif column_type in (int, float):
-        return str(value)
-
-    # Handle boolean types
+    # Handle boolean types (must come before numeric since bool is subclass of int)
     elif column_type is bool:
         return "TRUE" if value else "FALSE"
 
+    # Handle numeric types
+    elif column_type in (int, float) or (
+        inspect.isclass(column_type) and issubclass(column_type, (int, float))
+    ):
+        return str(value)
+
     # Handle date types
     elif column_type is date:
         if dialect == "bigquery":
@@ -508,7 +511,9 @@ def format_sql_value(value: Any, column_type: Type, dialect: str = "standard") -
                 return f"TIMESTAMP '{value}'"
 
     # Handle decimal types
-    elif column_type == Decimal:
+    elif column_type == Decimal or (
+        inspect.isclass(column_type) and issubclass(column_type, Decimal)
+    ):
         if dialect == "bigquery":
             # BigQuery needs explicit NUMERIC casting for decimals
             return f"NUMERIC '{value}'"