pytest-revealtype-injector 0.6.0__tar.gz → 0.6.1__tar.gz

{pytest_revealtype_injector-0.6.0 → pytest_revealtype_injector-0.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-revealtype-injector
-Version: 0.6.0
+Version: 0.6.1
 Summary: Pytest plugin for replacing reveal_type() calls inside test functions with static and runtime type checking result comparison, for confirming type annotation validity.
 Project-URL: homepage, https://github.com/abelcheung/pytest-revealtype-injector
 Author-email: Abel Cheung <abelcheung@gmail.com>

{pytest_revealtype_injector-0.6.0 → pytest_revealtype_injector-0.6.1}/src/pytest_revealtype_injector/__init__.py

@@ -1,3 +1,3 @@
 """Pytest plugin for replacing reveal_type() calls inside test functions with static and runtime type checking result comparison, for confirming type annotation validity."""  # noqa: E501
 
-__version__ = "0.6.0"
+__version__ = "0.6.1"

{pytest_revealtype_injector-0.6.0 → pytest_revealtype_injector-0.6.1}/src/pytest_revealtype_injector/adapter/mypy_.py

@@ -125,6 +125,10 @@ class NameCollector(NameCollectorBase):
     # something like "test_elem_class_lookup.FooClass@97".
     # Return only the left operand after processing.
     def visit_BinOp(self, node: ast.BinOp) -> ast.expr:
+        if isinstance(node.op, ast.BitOr):  # union
+            node.left = self.visit(node.left)
+            node.right = self.visit(node.right)
+            return node
         if isinstance(node.op, ast.MatMult) and isinstance(node.right, ast.Constant):
             return cast("ast.expr", self.visit(node.left))
         # For expression that haven't been accounted for, just don't