fix(python): boolean mapping

Author: paulolieuthier

bindings/python/src/lib.rs

@@ -85,6 +85,10 @@ fn from(ob: &Bound<'_, PyAny>) -> Result<Value, PyErr> {
     else if let Ok(s) = ob.extract::<String>() {
         s.into()
     }
+    // Boolean
+    else if let Ok(b) = ob.extract::<bool>() {
+        b.into()
+    }
     // Numeric
     else if let Ok(v) = ob.extract::<i64>() {
         v.into()
@@ -93,10 +97,6 @@ fn from(ob: &Bound<'_, PyAny>) -> Result<Value, PyErr> {
     } else if let Ok(v) = ob.extract::<f64>() {
         v.into()
     }
-    // Boolean
-    else if let Ok(b) = ob.extract::<bool>() {
-        b.into()
-    }
     // None
     else if ob.downcast::<PyNone>().is_ok() {
         Value::Null

bindings/python/test.py

@@ -387,3 +387,29 @@ def modify_set(st):
     assert st == {10, 12}, f"Unexpected set: {st}"
 
 test_extension_types()
+
+def test_boolean_mapping():
+    rego = regorus.Engine()
+    rego.add_policy("demo",
+        """
+        package demo
+
+        result_b := data.a if {
+            data.a == true
+        }
+
+        result_i := data.b if {
+            data.b == 1
+        }
+        """)
+    rego.add_data({"a": True, "b": 1})
+
+    result_b = rego.eval_rule("data.demo.result_b")
+    assert isinstance(result_b, bool), f"Expected bool, got {type(result_b)}"
+    assert result_b, f"Unexpected result for 'result_b': {result_b}"
+
+    result_i = rego.eval_rule("data.demo.result_i")
+    assert isinstance(result_i, float), f"Expected float, got {type(result_i)}"
+    assert result_i == 1, f"Unexpected result for 'result_i': {result_i}"
+
+test_boolean_mapping()