def setUp(self): self.obj = CFTester()
class CFTesterTest(unittest.TestCase): def setUp(self): self.obj = CFTester() def test_run(self): self.obj.set_command1('python test1.py') self.obj.set_command2('python test2.py') def f(): yield "0 0 0" yield "1 2 3" yield "-1 -3 -5" yield "-2 -3 -5" self.obj.set_generator(f) self.assertEqual(0,self.obj.run()) def test_generate_input(self): f = 'function' self.assertEqual(0,self.obj.set_generator(f)) self.assertEqual(f,self.obj.get_generator()) def test_get_command1(self): self.obj.set_command1('python test1.py') self.assertEqual('python test1.py',self.obj.get_command1()) def test_set_command1(self): self.assertEqual(0,self.obj.set_command1('python test1.py')) def test_get_command2(self): self.obj.set_command2('python test2.py') self.assertEqual('python test2.py',self.obj.get_command2()) def test_set_command2(self): self.assertEqual(0,self.obj.set_command2('python test2.py')) def test_run_test1(self): self.assertEqual(0,self.obj.set_command1('python test1.py')) self.assertEqual('0.0',self.obj.run_test1('0 0 0')) self.assertEqual('5.0',self.obj.run_test1('3 4 0')) def test_run_test2(self): self.assertEqual(0,self.obj.set_command2('python test2.py')) self.assertEqual('0.0',self.obj.run_test2('0 0 0')) self.assertEqual('5.0',self.obj.run_test2('3 4 0'))