Try switching entirely to rust

Author: saulshanabrook

.flake8

@@ -1,4 +1,4 @@
 ; https://black.readthedocs.io/en/stable/guides/using_black_with_other_tools.html#flake8
 [flake8]
 max-line-length = 88
-extend-ignore = E203,E501,F405,F403,E302
+extend-ignore = E203,E501,F405,F403,E302,E305,F821

.github/workflows/CI.yml

@@ -38,7 +38,7 @@ jobs:
           python-version: "3.10"
       - uses: actions/checkout@v2
       - run: pip install -e . mypy
-      - run: python -m mypy.stubtest egg_smol.bindings
+      - run: python -m mypy.stubtest egg_smol.bindings  --allowlist stubtest_allow
   docs:
     runs-on: ubuntu-latest
     steps:

python/egg_smol/bindings.pyi

@@ -3,7 +3,119 @@ from typing import Optional
 
 from typing_extensions import final
 
-from .bindings_py import Expr, Fact_, FunctionDecl, Rewrite
+@final
+class EGraph:
+    def parse_and_run_program(self, input: str) -> list[str]: ...
+    def declare_constructor(self, variant: Variant, sort: str) -> None: ...
+    def declare_sort(self, name: str) -> None: ...
+    def declare_function(self, decl: FunctionDecl) -> None: ...
+    def define(self, name: str, expr: _Expr, cost: Optional[int] = None) -> None: ...
+    def add_rewrite(self, rewrite: Rewrite) -> str: ...
+    def run_rules(self, limit: int) -> tuple[timedelta, timedelta, timedelta]: ...
+    def check_fact(self, fact: _Fact) -> None: ...
+
+@final
+class EggSmolError(Exception):
+    context: str
+
+@final
+class Int:
+    def __init__(self, value: int) -> None: ...
+    value: int
+
+@final
+class String:
+    def __init__(self, value: str) -> None: ...
+    value: str
+
+@final
+class Unit:
+    def __init__(self) -> None: ...
+
+_Literal = Int | String | Unit
+
+@final
+class Lit:
+    def __init__(self, value: _Literal) -> None: ...
+    value: _Literal
+
+@final
+class Var:
+    def __init__(self, name: str) -> None: ...
+    name: str
+
+@final
+class Call:
+    def __init__(self, name: str, args: list[_Expr]) -> None: ...
+    name: str
+    args: list[_Expr]
+
+_Expr = Lit | Var | Call
+
+@final
+class Eq:
+    def __init__(self, exprs: list[_Expr]) -> None: ...
+    exprs: list[_Expr]
+
+@final
+class Fact:
+    def __init__(self, expr: _Expr) -> None: ...
+    expr: _Expr
+
+_Fact = Fact | Eq
+
+@final
+class Define:
+    def __init__(self, lhs: str, rhs: _Expr) -> None: ...
+    lhs: str
+    rhs: _Expr
+
+@final
+class Set:
+    def __init__(self, lhs: str, args: list[_Expr], rhs: _Expr) -> None: ...
+    lhs: str
+    args: list[_Expr]
+    rhs: _Expr
+
+@final
+class Delete:
+    sym: str
+    args: list[_Expr]
+    def __init__(self, sym: str, args: list[_Expr]) -> None: ...
+
+@final
+class Union:
+    def __init__(self, lhs: _Expr, rhs: _Expr) -> None: ...
+    lhs: _Expr
+    rhs: _Expr
+
+@final
+class Panic:
+    def __init__(self, msg: str) -> None: ...
+    msg: str
+
+@final
+class Expr_:
+    def __init__(self, expr: _Expr) -> None: ...
+    expr: _Expr
+
+_Action = Define | Set | Delete | Union | Panic | Expr_
+
+@final
+class FunctionDecl:
+    name: str
+    schema: Schema
+    default: Optional[_Expr]
+    merge: Optional[_Expr]
+    cost: Optional[int]
+    def __init__(
+        self,
+        name: str,
+        schema: Schema,
+        default: Optional[_Expr],
+        merge: Optional[_Expr],
+        cost: Optional[int] = None,
+    ) -> None: ...
 
 @final
 class Variant:
@@ -15,16 +127,23 @@ class Variant:
     cost: Optional[int]
 
 @final
-class EGraph:
-    def parse_and_run_program(self, input: str) -> list[str]: ...
-    def declare_constructor(self, variant: Variant, sort: str) -> None: ...
-    def declare_sort(self, name: str) -> None: ...
-    def declare_function(self, decl: FunctionDecl) -> None: ...
-    def define(self, name: str, expr: Expr, cost: Optional[int] = None) -> None: ...
-    def add_rewrite(self, rewrite: Rewrite) -> str: ...
-    def run_rules(self, limit: int) -> tuple[timedelta, timedelta, timedelta]: ...
-    def check_fact(self, fact: Fact_) -> None: ...
+class Schema:
+    input: list[str]
+    output: str
+    def __init__(self, input: list[str], output: str) -> None: ...
 
 @final
-class EggSmolError(Exception):
-    context: str
+class Rule:
+    head: list[_Action]
+    body: list[_Fact]
+    def __init__(self, head: list[_Action], body: list[_Fact]) -> None: ...
+
+@final
+class Rewrite:
+    lhs: _Expr
+    rhs: _Expr
+    conditions: list[_Fact]
+
+    def __init__(
+        self, lhs: _Expr, rhs: _Expr, conditions: list[_Fact] = []
+    ) -> None: ...

python/egg_smol/bindings_py.py

@@ -2,7 +2,6 @@
 
 import pytest
 from egg_smol.bindings import *
-from egg_smol.bindings_py import *
 
 
 class TestEGraph: