def test_class_def(self): output = StringIO.StringIO() f = Formatter() f.gen_class_def(out_file=output, class_name='ATest', description='Some description') output_buff = output.getvalue() self.assertTrue('ATest' in output_buff) self.assertTrue('Some description' in output_buff)
def test_gen_unittest_method(self): output = StringIO.StringIO() f = Formatter() f.gen_unittest_method(out_file=output, method_name='some_method') output_buff = output.getvalue() self.assertTrue('some_method' in output_buff)
def test_module_header(self): output = StringIO.StringIO() f = Formatter() f.gen_test_module_header(out_file=output, module_name='AModule') output_buff = output.getvalue() self.assertTrue('AModule' in output_buff)
def test_class_close(self): output = StringIO.StringIO() f = Formatter() f.gen_class_close(out_file=output, class_name='ATest') output_buff = output.getvalue() self.assertTrue('ATest' in output_buff)