def test_changing_the_module_and_class_methods(self): code = dedent("""\ class C(object): def a_func(self): return 1 def another_func(self): pass """) self.mod.write(code) replacer = MethodObject(self.project, self.mod, code.index("func")) self.project.do(replacer.get_changes("_New")) expected = dedent("""\ class C(object): def a_func(self): return _New(self)() def another_func(self): pass class _New(object): def __init__(self, host): self.self = host def __call__(self): return 1 """) self.assertEqual(expected, self.mod.read())
def test_changing_the_module(self): code = 'def func():\n return 1\n' self.mod.write(code) replacer = MethodObject(self.project, self.mod, code.index('func')) self.project.do(replacer.get_changes('_New')) expected = 'def func():\n' \ ' return _New()()\n\n\n' \ 'class _New(object):\n\n' \ ' def __call__(self):\n' \ ' return 1\n' self.assertEquals(expected, self.mod.read())
def test_changing_the_module(self): code = dedent("""\ def func(): return 1 """) self.mod.write(code) replacer = MethodObject(self.project, self.mod, code.index("func")) self.project.do(replacer.get_changes("_New")) expected = dedent("""\ def func(): return _New()() class _New(object): def __call__(self): return 1 """) self.assertEqual(expected, self.mod.read())
def test_changing_the_module_and_class_methods(self): code = 'class C(object):\n\n' \ ' def a_func(self):\n' \ ' return 1\n\n' \ ' def another_func(self):\n' \ ' pass\n' self.mod.write(code) replacer = MethodObject(self.project, self.mod, code.index('func')) self.project.do(replacer.get_changes('_New')) expected = 'class C(object):\n\n' \ ' def a_func(self):\n' \ ' return _New(self)()\n\n' \ ' def another_func(self):\n' \ ' pass\n\n\n' \ 'class _New(object):\n\n' \ ' def __init__(self, host):\n' \ ' self.self = host\n\n' \ ' def __call__(self):\n' \ ' return 1\n' self.assertEquals(expected, self.mod.read())