Fix format_threshold to cast infinity values in query

Author: Benjamin Gutzmann

CHANGELOG.md

@@ -20,6 +20,7 @@ Types of changes:
 
 - Fix query wrapping
 - Fix date column type casting in SQL queries to handle TIMESTAMP values correctly
+- Fix format_threshold to cast infinity values in query
 
 ## [0.6.0] - 2026-01-07
 

src/koality/executor.py

@@ -425,8 +425,8 @@ def load_to_database(self) -> None:
                         '{row[identifier_column]}',
                         {f"'{row['COLUMN']}'" if row["COLUMN"] is not None else "NULL"},
                         {row["VALUE"] if row["VALUE"] is not None else "NULL"},
-                        {format_threshold(row["LOWER_THRESHOLD"])},
-                        {format_threshold(row["UPPER_THRESHOLD"])},
+                        {format_threshold(row["LOWER_THRESHOLD"]).format(numeric_type=DATA_TYPES["NUMERIC"][self.database_provider.type])},
+                        {format_threshold(row["UPPER_THRESHOLD"]).format(numeric_type=DATA_TYPES["NUMERIC"][self.database_provider.type])},
                         '{row["RESULT"]}',
                         '{row["INSERT_TIMESTAMP"]}'
                     )

src/koality/utils.py

@@ -199,6 +199,6 @@ def format_threshold(value: float | None) -> str:
         return "NULL"
     if math.isinf(value):
         if value > 0:
-            return "'+Infinity'"
-        return "'-Infinity'"
+            return "CAST('+Infinity' AS {numeric_type})"
+        return "CAST('-Infinity' AS {numeric_type})"
     return str(value)

tests/unit/test_utils.py

@@ -150,8 +150,8 @@ def test_special_characters_in_value(self) -> None:
     ("value", "expected"),
     [
         (None, "NULL"),
-        (float("inf"), "'+Infinity'"),
-        (float("-inf"), "'-Infinity'"),
+        (float("inf"), "CAST('+Infinity' AS {numeric_type})"),
+        (float("-inf"), "CAST('-Infinity' AS {numeric_type})"),
         (0.0, "0.0"),
         (42.5, "42.5"),
         (-10.25, "-10.25"),