def test_get_name_for_factories(self): code = 'class C(object):\n pass\n' mod = testutils.create_module(self.project, 'mod') mod.write(code) factory = IntroduceFactory(self.project, mod, mod.read().index('C') + 1) self.assertEquals('C', factory.get_name())
def test_get_name_for_factories(self): code = dedent("""\ class C(object): pass """) mod = testutils.create_module(self.project, "mod") mod.write(code) factory = IntroduceFactory(self.project, mod, mod.read().index("C") + 1) self.assertEqual("C", factory.get_name())