Merge pull request #36 from metadsl/pre-commit-ci-update-config

[pre-commit.ci] pre-commit autoupdate

.pre-commit-config.yaml

@@ -13,7 +13,7 @@ repos:
     hooks:
       - id: black
   - repo: https://github.com/pycqa/flake8
-    rev: 6.0.0
+    rev: 6.1.0
     hooks:
       - id: flake8
   - repo: local

python/egglog/runtime.py

@@ -94,7 +94,7 @@ def _resolve_literal(tp: TypeOrVarRef, arg: object) -> RuntimeExpr:
     else:
         arg_type = type(arg)
         # If this value has a custom metaclass, let's use that as our index instead of the type
-        if type(arg_type) != type:
+        if type(arg_type) != type:  # noqa: E721
             arg_type = type(arg_type)
 
     # If we have any type variables, dont bother trying to resolve the literal, just return the arg

python/tests/test_convert.py

@@ -0,0 +1,35 @@
+from egglog import *
+
+
+def test_conversion_custom_metaclass():
+    class MyMeta(type):
+        pass
+
+    class MyType(metaclass=MyMeta):
+        pass
+
+    egraph = EGraph()
+
+    @egraph.class_
+    class MyTypeExpr(Expr):
+        def __init__(self):
+            ...
+
+    converter(MyMeta, MyTypeExpr, lambda x: MyTypeExpr())
+    assert expr_parts(convert(MyType(), MyTypeExpr)) == expr_parts(MyTypeExpr())
+
+
+def test_conversion():
+    egraph = EGraph()
+
+    class MyType:
+        pass
+
+    @egraph.class_
+    class MyTypeExpr(Expr):
+        def __init__(self):
+            ...
+
+    converter(MyType, MyTypeExpr, lambda x: MyTypeExpr())
+
+    assert expr_parts(convert(MyType(), MyTypeExpr)) == expr_parts(MyTypeExpr())