Add PyObject.to_int() method returning i64

python/egglog/builtins.py

@@ -1022,6 +1022,9 @@ def to_string(self) -> String: ...
     @method(egg_fn="py-to-bool")
     def to_bool(self) -> Bool: ...
 
+    @method(egg_fn="py-to-int")
+    def to_int(self) -> i64: ...
+
     @method(egg_fn="py-dict-update")
     def dict_update(self, *keys_and_values: object) -> PyObject: ...
 

python/tests/test_high_level.py

@@ -229,6 +229,9 @@ def test_from_string(self):
     def test_to_string(self):
         EGraph().check(PyObject("foo").to_string() == String("foo"))
 
+    def test_to_int(self):
+        EGraph().check(PyObject(42).to_int() == i64(42))
+
     def test_dict_update(self):
         original_d = {"foo": "bar"}
         res = EGraph().extract(PyObject(original_d).dict_update("foo", "baz")).value

src/py_object_sort.rs

@@ -170,6 +170,15 @@ impl BaseSort for PyObjectSort {
                 }
             }
         );
+        // (py-to-int <obj>)
+        add_primitive!(
+            eg,
+            "py-to-int" = |x: PyPickledValue| -?> i64 {
+                {
+                    attach("py-to-int", move |py| load(py, &x)?.extract())
+                }
+            }
+        );
         // (py-from-string <str>)
         add_primitive!(
             eg,