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)
