sql-testing-library 0.8.0__tar.gz → 0.9.0__tar.gz

{sql_testing_library-0.8.0 → sql_testing_library-0.9.0}/CHANGELOG.md

@@ -5,6 +5,12 @@ 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.9.0 (2025-06-10)
+
+### Feat
+
+- **redshift**: add support for map datatype for redshift (#98)
+
 ## 0.8.0 (2025-06-09)
 
 ### Feat

{sql_testing_library-0.8.0 → sql_testing_library-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sql-testing-library
-Version: 0.8.0
+Version: 0.9.0
 Summary: A powerful Python framework for unit testing SQL queries across BigQuery, Snowflake, Redshift, Athena, and Trino with mock data
 License: MIT
 Keywords: sql,testing,unit-testing,mock-data,database-testing,bigquery,snowflake,redshift,athena,trino,data-engineering,etl-testing,sql-validation,query-testing
@@ -69,6 +69,7 @@ A powerful Python framework for unit testing SQL queries with mock data injectio
 [![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)
 ![python version](https://img.shields.io/badge/python-3.9%2B-yellowgreen)
+[![Documentation](https://img.shields.io/badge/docs-GitHub%20Pages-blue)](https://gurmeetsaran.github.io/sqltesting/)
 
 ## 🎯 Motivation
 
@@ -135,8 +136,8 @@ The library supports different data types across database engines. All checkmark
 | **Integer Array** | `List[int]` | ✅ | ✅ | ✅ | ✅ | ✅ |
 | **Decimal Array** | `List[Decimal]` | ✅ | ✅ | ✅ | ✅ | ✅ |
 | **Optional Array** | `Optional[List[T]]` | ✅ | ✅ | ✅ | ✅ | ✅ |
+| **Map/Object** | `Dict[K, V]` | ❌ | ✅ | ✅ | ✅ | ❌ |
 | **Struct/Record** | `dict`/`dataclass` | ❌ | ❌ | ❌ | ❌ | ❌ |
-| **Map/Object** | `Dict[K, V]` | ❌ | ✅ | ❌ | ✅ | ❌ |
 | **Nested Arrays** | `List[List[T]]` | ❌ | ❌ | ❌ | ❌ | ❌ |
 | **JSON Objects** | `JSON` | ❌ | ❌ | ❌ | ❌ | ❌ |
 
@@ -144,7 +145,7 @@ The library supports different data types across database engines. All checkmark
 
 - **BigQuery**: NULL arrays become empty arrays `[]`; uses scientific notation for large decimals
 - **Athena**: 256KB query size limit; supports arrays and maps using `ARRAY[]` and `MAP(ARRAY[], ARRAY[])` syntax
-- **Redshift**: Arrays implemented via JSON parsing; 16MB query size limit
+- **Redshift**: Arrays and maps implemented via SUPER type (JSON parsing); 16MB query size limit
 - **Trino**: Memory catalog for testing; excellent decimal precision; supports arrays and maps
 - **Snowflake**: Column names normalized to lowercase; 1MB query size limit
 

{sql_testing_library-0.8.0 → sql_testing_library-0.9.0}/README.md

@@ -12,6 +12,7 @@ A powerful Python framework for unit testing SQL queries with mock data injectio
 [![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)
 ![python version](https://img.shields.io/badge/python-3.9%2B-yellowgreen)
+[![Documentation](https://img.shields.io/badge/docs-GitHub%20Pages-blue)](https://gurmeetsaran.github.io/sqltesting/)
 
 ## 🎯 Motivation
 
@@ -78,8 +79,8 @@ The library supports different data types across database engines. All checkmark
 | **Integer Array** | `List[int]` | ✅ | ✅ | ✅ | ✅ | ✅ |
 | **Decimal Array** | `List[Decimal]` | ✅ | ✅ | ✅ | ✅ | ✅ |
 | **Optional Array** | `Optional[List[T]]` | ✅ | ✅ | ✅ | ✅ | ✅ |
+| **Map/Object** | `Dict[K, V]` | ❌ | ✅ | ✅ | ✅ | ❌ |
 | **Struct/Record** | `dict`/`dataclass` | ❌ | ❌ | ❌ | ❌ | ❌ |
-| **Map/Object** | `Dict[K, V]` | ❌ | ✅ | ❌ | ✅ | ❌ |
 | **Nested Arrays** | `List[List[T]]` | ❌ | ❌ | ❌ | ❌ | ❌ |
 | **JSON Objects** | `JSON` | ❌ | ❌ | ❌ | ❌ | ❌ |
 
@@ -87,7 +88,7 @@ The library supports different data types across database engines. All checkmark
 
 - **BigQuery**: NULL arrays become empty arrays `[]`; uses scientific notation for large decimals
 - **Athena**: 256KB query size limit; supports arrays and maps using `ARRAY[]` and `MAP(ARRAY[], ARRAY[])` syntax
-- **Redshift**: Arrays implemented via JSON parsing; 16MB query size limit
+- **Redshift**: Arrays and maps implemented via SUPER type (JSON parsing); 16MB query size limit
 - **Trino**: Memory catalog for testing; excellent decimal precision; supports arrays and maps
 - **Snowflake**: Column names normalized to lowercase; 1MB query size limit
 

{sql_testing_library-0.8.0 → sql_testing_library-0.9.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "sql-testing-library"
-version = "0.8.0"
+version = "0.9.0"
 description = "A powerful Python framework for unit testing SQL queries across BigQuery, Snowflake, Redshift, Athena, and Trino 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.8.0 → sql_testing_library-0.9.0}/src/sql_testing_library/_adapters/redshift.py

@@ -32,6 +32,32 @@ class RedshiftTypeConverter(BaseTypeConverter):
 
     def convert(self, value: Any, target_type: Type) -> Any:
         """Convert Redshift result value to target type."""
+        # Handle None/NULL values first
+        if value is None:
+            return None
+
+        # Handle Optional types
+        if self.is_optional_type(target_type):
+            if value is None:
+                return None
+            target_type = self.get_optional_inner_type(target_type)
+
+        # Handle dict/map types from Redshift SUPER columns
+        if hasattr(target_type, "__origin__") and target_type.__origin__ is dict:
+            # Redshift returns SUPER types as Python dicts already
+            if isinstance(value, dict):
+                return value
+            # If it's a string (shouldn't happen with psycopg2), parse it
+            elif isinstance(value, str):
+                import json
+
+                try:
+                    return json.loads(value)
+                except json.JSONDecodeError:
+                    return {}
+            else:
+                return {}
+
         # Redshift returns proper Python types in most cases, so use base converter
         return super().convert(value, target_type)
 
@@ -190,7 +216,11 @@ class RedshiftAdapter(DatabaseAdapter):
                     if non_none_types:
                         col_type = non_none_types[0]
 
-                redshift_type = type_mapping.get(col_type, "VARCHAR")
+                # Handle dict/map types
+                if hasattr(col_type, "__origin__") and col_type.__origin__ is dict:
+                    redshift_type = "SUPER"
+                else:
+                    redshift_type = type_mapping.get(col_type, "VARCHAR")
                 column_defs.append(f'"{col_name}" {redshift_type}')
 
             columns_sql = ",\n  ".join(column_defs)

{sql_testing_library-0.8.0 → sql_testing_library-0.9.0}/src/sql_testing_library/_sql_utils.py

@@ -1,8 +1,19 @@
 """SQL utility functions for escaping and formatting values."""
 
+import json
+from decimal import Decimal
 from typing import Any, Type
 
 
+class DecimalEncoder(json.JSONEncoder):
+    """JSON encoder that handles Decimal types."""
+
+    def default(self, o):
+        if isinstance(o, Decimal):
+            return float(o)
+        return super().default(o)
+
+
 def escape_sql_string(value: str) -> str:
     """
     Escape a string value for SQL using standard SQL escaping rules.
@@ -153,6 +164,9 @@ def format_sql_value(value: Any, column_type: Type, dialect: str = "standard") -
                 sql_key_type = get_sql_type(key_type)
                 sql_value_type = get_sql_type(value_type)
                 return f"CAST(NULL AS MAP({sql_key_type}, {sql_value_type}))"
+            elif dialect == "redshift":
+                # Redshift SUPER type handles NULL maps
+                return "NULL::SUPER"
             else:
                 return "NULL"
 
@@ -220,9 +234,6 @@ def format_sql_value(value: Any, column_type: Type, dialect: str = "standard") -
         elif dialect == "redshift":
             # Redshift uses JSON-like syntax for SUPER arrays
             # Format elements as JSON (double quotes for strings)
-            import json
-            from decimal import Decimal
-
             # Convert elements to JSON-serializable types
             json_elements = []
             for element in value:
@@ -274,9 +285,13 @@ def format_sql_value(value: Any, column_type: Type, dialect: str = "standard") -
                 keys.append(format_sql_value(k, key_type, dialect))
                 values.append(format_sql_value(v, value_type, dialect))
             return f"MAP(ARRAY[{', '.join(keys)}], ARRAY[{', '.join(values)}])"
+        elif dialect == "redshift":
+            # Redshift uses SUPER type with JSON-like syntax for maps
+            json_str = json.dumps(value, cls=DecimalEncoder)
+            return f"JSON_PARSE('{json_str}')"
         else:
             # Other databases don't have native map support yet
-            # Could potentially use JSON for BigQuery, Redshift, Snowflake
+            # Could potentially use JSON for BigQuery, Snowflake
             raise NotImplementedError(f"Map type not yet supported for dialect: {dialect}")
 
     # Handle string types