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())
def _introduce_factory(self, resource, offset, *args, **kwds): factory_introducer = IntroduceFactory(self.project, resource, offset) changes = factory_introducer.get_changes(*args, **kwds) self.project.do(changes)