datajunction 0.0.150__tar.gz → 0.0.152__tar.gz

{datajunction-0.0.150 → datajunction-0.0.152}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datajunction
-Version: 0.0.150
+Version: 0.0.152
 Summary: DataJunction client library for connecting to a DataJunction server
 Project-URL: repository, https://github.com/DataJunction/dj
 Author-email: DataJunction Authors <yian.shang@gmail.com>

{datajunction-0.0.150 → datajunction-0.0.152}/datajunction/__about__.py

@@ -2,4 +2,4 @@
 Version for Hatch
 """
 
-__version__ = "0.0.150"
+__version__ = "0.0.152"

{datajunction-0.0.150 → datajunction-0.0.152}/tests/conftest.py

@@ -215,17 +215,21 @@ def module__query_service_client(
             Dict[str, str]
         ] = None,
     ) -> QueryWithResults:
-        normalized_query = (
-            query_create.submitted_query.strip()
-            .replace('"', "")
-            .replace("\n", "")
-            .replace(" ", "")
-        )
-        # Strip LIMIT clause for matching (allows queries with/without LIMIT to match)
-        normalized_query = re.sub(r"LIMIT\d+$", "", normalized_query)
+        def _normalize(s: str) -> str:
+            out = (
+                s.strip()
+                .replace('"', "")
+                .replace("\n", "")
+                .replace("\t", "")
+                .replace(" ", "")
+            )
+            return re.sub(r"LIMIT\d+$", "", out)
+
+        normalized_query = _normalize(query_create.submitted_query)
 
-        if normalized_query in QUERY_DATA_MAPPINGS:
-            results = QUERY_DATA_MAPPINGS[normalized_query]
+        normalized_mappings = {_normalize(k): v for k, v in QUERY_DATA_MAPPINGS.items()}
+        if normalized_query in normalized_mappings:
+            results = normalized_mappings[normalized_query]
         else:
             # SQL-agnostic fallback: return a generic successful result.
             # Client tests should not depend on the exact SQL generated — they