64 lines
2 KiB
Python
64 lines
2 KiB
Python
|
"""Visitor classes pulled out from different tests
|
||
|
|
|
||
|
|
These are here because we don't currently support having interpreted
|
||
|
|
classes subtype compiled ones but pytest grabs the python file
|
||
|
|
even if the test was compiled.
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from mypy.nodes import AssignmentStmt, CallExpr, Expression, IntExpr, NameExpr, Node, TypeVarExpr
|
||
|
|
from mypy.traverser import TraverserVisitor
|
||
|
|
from mypy.treetransform import TransformVisitor
|
||
|
|
from mypy.types import Type
|
||
|
|
|
||
|
|
|
||
|
|
# from testtypegen
|
||
|
|
class SkippedNodeSearcher(TraverserVisitor):
|
||
|
|
def __init__(self) -> None:
|
||
|
|
self.nodes: set[Node] = set()
|
||
|
|
self.ignore_file = False
|
||
|
|
|
||
|
|
def visit_assignment_stmt(self, s: AssignmentStmt) -> None:
|
||
|
|
if s.type or ignore_node(s.rvalue):
|
||
|
|
for lvalue in s.lvalues:
|
||
|
|
if isinstance(lvalue, NameExpr):
|
||
|
|
self.nodes.add(lvalue)
|
||
|
|
super().visit_assignment_stmt(s)
|
||
|
|
|
||
|
|
def visit_name_expr(self, n: NameExpr) -> None:
|
||
|
|
if self.ignore_file:
|
||
|
|
self.nodes.add(n)
|
||
|
|
super().visit_name_expr(n)
|
||
|
|
|
||
|
|
def visit_int_expr(self, n: IntExpr) -> None:
|
||
|
|
if self.ignore_file:
|
||
|
|
self.nodes.add(n)
|
||
|
|
super().visit_int_expr(n)
|
||
|
|
|
||
|
|
|
||
|
|
def ignore_node(node: Expression) -> bool:
|
||
|
|
"""Return True if node is to be omitted from test case output."""
|
||
|
|
|
||
|
|
# We want to get rid of object() expressions in the typing module stub
|
||
|
|
# and also TypeVar(...) expressions. Since detecting whether a node comes
|
||
|
|
# from the typing module is not easy, we just to strip them all away.
|
||
|
|
if isinstance(node, TypeVarExpr):
|
||
|
|
return True
|
||
|
|
if isinstance(node, NameExpr) and node.fullname == "builtins.object":
|
||
|
|
return True
|
||
|
|
if isinstance(node, NameExpr) and node.fullname == "builtins.None":
|
||
|
|
return True
|
||
|
|
if isinstance(node, CallExpr) and (ignore_node(node.callee) or node.analyzed):
|
||
|
|
return True
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# from testtransform
|
||
|
|
class TypeAssertTransformVisitor(TransformVisitor):
|
||
|
|
def type(self, type: Type) -> Type:
|
||
|
|
assert type is not None
|
||
|
|
return type
|