move type tests to own file

Author: saulshanabrook

python/egg_smol/type_constraint_solver.py

@@ -69,31 +69,3 @@ def _subtitute_typevars(self, tp: TypeOrVarRef) -> JustTypeRef:
                 tp.name,
                 tuple(self._subtitute_typevars(arg) for arg in tp.args),
             )
-
-
-def test_type_inference() -> None:
-    import pytest
-
-    i64 = TypeRefWithVars("i64")
-    unit = TypeRefWithVars("Unit")
-    K, V = ClassTypeVarRef(0), ClassTypeVarRef(1)
-    map = TypeRefWithVars("Map", (K, V))
-    map_i64_unit = TypeRefWithVars("Map", (i64, unit))
-
-    cs = TypeConstraintSolver()
-    assert cs.infer_return_type([i64], i64, [i64.to_just()]) == i64.to_just()
-    with pytest.raises(TypeError):
-        cs.infer_return_type([i64], i64, [unit.to_just()])
-    with pytest.raises(TypeError):
-        cs.infer_return_type([], i64, [unit.to_just()])
-
-    assert cs.infer_return_type([map, K], V, [map_i64_unit.to_just(), i64.to_just()]) == unit.to_just()
-
-    with pytest.raises(TypeError):
-        cs.infer_return_type([map, K], V, [map_i64_unit.to_just(), unit.to_just()])
-
-    bound_cs = TypeConstraintSolver.from_type_parameters([i64.to_just(), unit.to_just()])
-    assert bound_cs.infer_return_type([K], V, [i64.to_just()]) == unit.to_just()
-
-    with pytest.raises(TypeError):
-        bound_cs.infer_return_type([K], V, [unit.to_just()])

python/tests/test_type_constraint_solver.py

@@ -0,0 +1,31 @@
+from __future__ import annotations
+
+import pytest
+from egg_smol.declarations import *
+from egg_smol.type_constraint_solver import *
+
+
+def test_type_inference() -> None:
+    i64 = TypeRefWithVars("i64")
+    unit = TypeRefWithVars("Unit")
+    K, V = ClassTypeVarRef(0), ClassTypeVarRef(1)
+    map = TypeRefWithVars("Map", (K, V))
+    map_i64_unit = TypeRefWithVars("Map", (i64, unit))
+
+    cs = TypeConstraintSolver()
+    assert cs.infer_return_type([i64], i64, [i64.to_just()]) == i64.to_just()
+    with pytest.raises(TypeError):
+        cs.infer_return_type([i64], i64, [unit.to_just()])
+    with pytest.raises(TypeError):
+        cs.infer_return_type([], i64, [unit.to_just()])
+
+    assert cs.infer_return_type([map, K], V, [map_i64_unit.to_just(), i64.to_just()]) == unit.to_just()
+
+    with pytest.raises(TypeError):
+        cs.infer_return_type([map, K], V, [map_i64_unit.to_just(), unit.to_just()])
+
+    bound_cs = TypeConstraintSolver.from_type_parameters([i64.to_just(), unit.to_just()])
+    assert bound_cs.infer_return_type([K], V, [i64.to_just()]) == unit.to_just()
+
+    with pytest.raises(TypeError):
+        bound_cs.infer_return_type([K], V, [unit.to_just()])