def test_append_class(self): ut_class = UTClassDetails(class_name='SomeClass', base_class='TestCase') ut_class.method_names = ['first_feature', 'second_feature'] result = self.container.append_class(ut_class=ut_class) self.assertTrue(result) self.assertTrue(self.container.save_module('output4.py'))
def test_add_method(self): details = UTClassDetails(class_name='SomeClass') details.add_method('foo') self.assertEqual(details.method_names, ['foo'])