def testOptionalArguments(self):
src = textwrap.dedent("""
def foo(a: A, ...) -> Z
def foo(a: A) -> Z
def foo(a: A, b: B) -> Z
def foo(a: A, b: B, ...) -> Z
def foo() -> Z
""")
expected = textwrap.dedent("""
def foo(a: A, ...) -> Z
def foo() -> Z
""")
new_src = self.ApplyVisitorToString(src, optimize.ApplyOptionalArguments())
self.AssertSourceEquals(new_src, expected)
def test_optional_arguments(self):
src = pytd_src("""
def foo(a: A, *args) -> Z: ...
def foo(a: A) -> Z: ...
def foo(a: A, b: B) -> Z: ...
def foo(a: A, b: B, *args) -> Z: ...
def foo() -> Z: ...
""")
expected = pytd_src("""
def foo(a: A, *args) -> Z: ...
def foo() -> Z: ...
""")
new_src = self.ApplyVisitorToString(src,
optimize.ApplyOptionalArguments())
self.AssertSourceEquals(new_src, expected)