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'])
