def testBuiltinSuperClasses(self): src = textwrap.dedent(""" def f(x: list or object, y: int or float) -> int or bool """) expected = textwrap.dedent(""" def f(x, y) -> int """) visitor = optimize.FindCommonSuperClasses(use_abcs=False) new_src = self.ApplyVisitorToString(src, visitor) self.AssertSourceEquals(new_src, expected)
def testABCSuperClasses(self): src = textwrap.dedent(""" def f(x: list or tuple, y: frozenset or set) -> int or float def g(x: dict or Mapping, y: complex or int) -> set or dict or tuple or Container def h(x) -> ? """) expected = textwrap.dedent(""" def f(x: Sequence, y: Set) -> Real def g(x: Mapping, y: Complex) -> Container def h(x) -> ? """) visitor = optimize.FindCommonSuperClasses(use_abcs=True) new_src = self.ApplyVisitorToString(src, visitor) self.AssertSourceEquals(new_src, expected)
def testBuiltinSuperClasses(self): src = textwrap.dedent(""" def f(x: list or object, y: int or float) -> int or bool """) expected = textwrap.dedent(""" def f(x, y) -> int """) b = builtins.GetBuiltinsPyTD() hierarchy = b.Visit(visitors.ExtractSuperClassesByName()) visitor = optimize.FindCommonSuperClasses( optimize.SuperClassHierarchy(hierarchy)) ast = self.ParseAndResolve(src) ast = ast.Visit(visitor) ast = ast.Visit(visitors.DropBuiltinPrefix()) self.AssertSourceEquals(ast, expected)
def testFindCommonSuperClasses(self): src = textwrap.dedent(""" x = ... # type: int or other.Bar """) expected = textwrap.dedent(""" x = ... # type: int or other.Bar """) ast = self.Parse(src) ast = ast.Visit(visitors.ReplaceTypes( {"other.Bar": pytd.LateType("other.Bar")})) hierarchy = ast.Visit(visitors.ExtractSuperClassesByName()) ast = ast.Visit(optimize.FindCommonSuperClasses( optimize.SuperClassHierarchy(hierarchy))) ast = ast.Visit(visitors.LateTypeToClassType()) self.AssertSourceEquals(ast, expected)
def test_find_common_superclasses(self): src = pytd_src(""" x = ... # type: Union[int, other.Bar] """) expected = pytd_src(""" x = ... # type: Union[int, other.Bar] """) ast = self.Parse(src) ast = ast.Visit( visitors.ReplaceTypes({"other.Bar": pytd.LateType("other.Bar")})) hierarchy = ast.Visit(visitors.ExtractSuperClassesByName()) ast = ast.Visit( optimize.FindCommonSuperClasses( optimize.SuperClassHierarchy(hierarchy))) ast = ast.Visit(visitors.LateTypeToClassType()) self.AssertSourceEquals(ast, expected)
def test_builtin_superclasses(self): src = pytd_src(""" def f(x: Union[list, object], y: Union[complex, memoryview]) -> Union[int, bool]: ... """) expected = pytd_src(""" def f(x: builtins.object, y: builtins.object) -> builtins.int: ... """) hierarchy = self.builtins.Visit(visitors.ExtractSuperClassesByName()) hierarchy.update( self.typing.Visit(visitors.ExtractSuperClassesByName())) visitor = optimize.FindCommonSuperClasses( optimize.SuperClassHierarchy(hierarchy)) ast = self.ParseAndResolve(src) ast = ast.Visit(visitor) ast = ast.Visit(visitors.CanonicalOrderingVisitor()) self.AssertSourceEquals(ast, expected)
def testBuiltinSuperClasses(self): src = textwrap.dedent(""" def f(x: list or object, y: complex or memoryview) -> int or bool """) expected = textwrap.dedent(""" def f(x: object, y: object) -> int """) hierarchy = self.builtins.Visit(visitors.ExtractSuperClassesByName()) hierarchy.update(self.typing.Visit(visitors.ExtractSuperClassesByName())) visitor = optimize.FindCommonSuperClasses( optimize.SuperClassHierarchy(hierarchy)) ast = self.ParseAndResolve(src) ast = ast.Visit(visitor) ast = ast.Visit(visitors.DropBuiltinPrefix()) ast = ast.Visit(visitors.CanonicalOrderingVisitor()) self.AssertSourceEquals(ast, expected)
def test_user_superclass_hierarchy(self): class_data = pytd_src(""" class AB: pass class EFG: pass class A(AB, EFG): pass class B(AB): pass class E(EFG, AB): pass class F(EFG): pass class G(EFG): pass """) src = pytd_src(""" from typing import Any def f(x: Union[A, B], y: A, z: B) -> Union[E, F, G]: ... def g(x: Union[E, F, G, B]) -> Union[E, F]: ... def h(x) -> Any: ... """) + class_data expected = pytd_src(""" from typing import Any def f(x: AB, y: A, z: B) -> EFG: ... def g(x: object) -> EFG: ... def h(x) -> Any: ... """) + class_data hierarchy = self.Parse(src).Visit(visitors.ExtractSuperClassesByName()) visitor = optimize.FindCommonSuperClasses( optimize.SuperClassHierarchy(hierarchy)) new_src = self.ApplyVisitorToString(src, visitor) self.AssertSourceEquals(new_src, expected)
def testUserSuperClassHierarchy(self): class_data = textwrap.dedent(""" class AB(object): pass class EFG(object): pass class A(AB, EFG): pass class B(AB): pass class E(EFG, AB): pass class F(EFG): pass class G(EFG): pass """) src = textwrap.dedent(""" def f(x: A or B, y: A, z: B) -> E or F or G def g(x: E or F or G or B) -> E or F def h(x) -> ? """) + class_data expected = textwrap.dedent(""" def f(x: AB, y: A, z: B) -> EFG def g(x) -> EFG def h(x) -> ? """) + class_data hierarchy = self.Parse(src).Visit( visitors.ExtractSuperClassesByName()) visitor = optimize.FindCommonSuperClasses( optimize.SuperClassHierarchy(hierarchy)) new_src = self.ApplyVisitorToString(src, visitor) self.AssertSourceEquals(new_src, expected)