def test_removing_keyword_arg2(self):
self.mod.write('def a_func(p1, *args, **kwds):\n pass\na_func(1)\n')
signature = ChangeSignature(self.project, self.mod,
self.mod.read().index('a_func') + 1)
signature.remove(2).do()
self.assertEquals('def a_func(p1, *args):\n pass\na_func(1)\n',
self.mod.read())
def test_removing_arguments_passed_as_keywords(self):
self.mod.write('def a_func(p1):\n pass\na_func(p1=1)\n')
signature = ChangeSignature(self.project, self.mod,
self.mod.read().index('a_func') + 1)
signature.remove(0).do()
self.assertEquals('def a_func():\n pass\na_func()\n',
self.mod.read())
def test_removing_arguments_star_args(self):
self.mod.write('def a_func(p1, *args):\n pass\na_func(1)\n')
signature = ChangeSignature(self.project, self.mod,
self.mod.read().index('a_func') + 1)
signature.remove(1).do()
self.assertEquals('def a_func(p1):\n pass\na_func(1)\n',
self.mod.read())
def test_removing_arguments_with_multiple_args(self):
self.mod.write('def a_func(p1, p2):\n pass\na_func(1, 2)\n')
signature = ChangeSignature(self.project, self.mod,
self.mod.read().index('a_func') + 1)
signature.remove(0).do()
self.assertEquals('def a_func(p2):\n pass\na_func(2)\n',
self.mod.read())
def test_differentiating_class_accesses_from_instance_accesses(self):
self.mod.write(
'class A(object):\n def a_func(self, param):\n pass\n'
'a_var = A()\nA.a_func(a_var, param=1)')
signature = ChangeSignature(self.project, self.mod,
self.mod.read().index('a_func') + 1)
signature.remove(1).do()
self.assertEquals(
'class A(object):\n def a_func(self):\n pass\n'
'a_var = A()\nA.a_func(a_var)', self.mod.read())