feat(transform): Add builtins

Author: ducdetronquito

magicparse/transform.py

@@ -84,6 +84,10 @@ def assert_positive[T: int | float | Decimal](value: T) -> T:
     return value
 
 
+def is_zero(value: int | float | Decimal) -> bool:
+    return value == 0
+
+
 def left_pad_zeroes(value: str, width: int) -> str:
     try:
         return value.zfill(width)
@@ -167,6 +171,7 @@ def get_builtin_functions() -> dict[str, Callable[..., Any]]:
             "assert_positive": assert_positive,
             "coalesce_numbers": coalesce_numbers,
             "divide": divide,
+            "is_zero": is_zero,
             "left_pad_zeroes": left_pad_zeroes,
             "length": length,
             "map_to": map_to,

tests/test_transform.py

@@ -33,6 +33,12 @@ def test_divide():
         Transform("$divide(1, 0)").evaluate({})
 
 
+def test_is_zero():
+    assert Transform(expression="$is_zero(1)").evaluate({}) is False
+    assert Transform(expression="$is_zero(0)").evaluate({}) is True
+    assert Transform(expression="$is_zero(input)").evaluate({"input": Decimal(0)}) is True
+
+
 def test_left_pad_zeroes():
     assert Transform('$left_pad_zeroes("", 5)').evaluate({}) == "00000"
     assert Transform('$left_pad_zeroes("111", 5)').evaluate({}) == "00111"