Add Matches.

Author: dnwpark

tests/test_type_eval.py

@@ -33,6 +33,7 @@
     IsSub,
     Iter,
     Length,
+    Matches,
     Member,
     Members,
     NewProtocol,
@@ -902,6 +903,99 @@ def test_never_is():
     assert d == _LiteralGeneric[True]
 
 
+def test_matches_01():
+    d = eval_typing(Matches[int, int])
+    assert d == _LiteralGeneric[True]
+
+    d = eval_typing(Matches[int, str])
+    assert d == _LiteralGeneric[False]
+
+    d = eval_typing(Matches[str, int])
+    assert d == _LiteralGeneric[False]
+
+
+def test_matches_02():
+    class A:
+        pass
+
+    class B(A):
+        pass
+
+    class C(B):
+        pass
+
+    class D(A):
+        pass
+
+    class X:
+        pass
+
+    d = eval_typing(Matches[A, A])
+    assert d == _LiteralGeneric[True]
+
+    d = eval_typing(Matches[A, B])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[B, A])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[B, C])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[C, B])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[C, D])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[D, C])
+    assert d == _LiteralGeneric[False]
+
+    d = eval_typing(Matches[A, X])
+    assert d == _LiteralGeneric[False]
+
+
+def test_matches_03():
+    class A[T]:
+        pass
+
+    class B[T](A[T]):
+        pass
+
+    class C(B[int]):
+        pass
+
+    class D(A[str]):
+        pass
+
+    class X:
+        pass
+
+    d = eval_typing(Matches[A[int], A[int]])
+    assert d == _LiteralGeneric[True]
+    d = eval_typing(Matches[A[int], A[str]])
+    assert d == _LiteralGeneric[True]
+
+    d = eval_typing(Matches[A[int], B[int]])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[B[int], A[int]])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[A[int], B[str]])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[B[str], A[int]])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[B[int], C])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[C, B[int]])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[B[str], C])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[C, B[str]])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[C, D])
+    assert d == _LiteralGeneric[False]
+    d = eval_typing(Matches[D, C])
+    assert d == _LiteralGeneric[False]
+
+    d = eval_typing(Matches[A[int], X])
+    assert d == _LiteralGeneric[False]
+
+
 def test_eval_iter_01():
     d = eval_typing(Iter[tuple[int, str]])
     assert tuple(d) == (int, str)

typemap/type_eval/_eval_operators.py

@@ -35,6 +35,7 @@
     Iter,
     Length,
     Lowercase,
+    Matches,
     Member,
     Members,
     NewProtocol,
@@ -246,6 +247,14 @@ def _eval_IsSubSimilar(lhs, rhs, *, ctx):
     return _LiteralGeneric[type_eval.issubsimilar(lhs, rhs)]
 
 
+@type_eval.register_evaluator(Matches)
+@_lift_evaluated
+def _eval_Matches(lhs, rhs, *, ctx):
+    return _LiteralGeneric[
+        type_eval.issubsimilar(lhs, rhs) and type_eval.issubsimilar(rhs, lhs)
+    ]
+
+
 def _eval_bool_tp(tp):
     if _typing_inspect.is_generic_alias(tp):
         if tp.__origin__ is typing.Literal:

typemap/typing.py

@@ -196,6 +196,11 @@ def IsSubSimilar(self, tps):
     return _BoolGenericAlias(self, tps)
 
 
+@_SpecialForm
+def Matches(self, tps):
+    return _BoolGenericAlias(self, tps)
+
+
 IsSub = IsSubSimilar