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

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

@@ -5,6 +5,18 @@ 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

{sql_testing_library-0.16.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.16.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

{sql_testing_library-0.16.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.16.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>"]

{sql_testing_library-0.16.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.16.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,
             )