def test_one_arg(self): for i in range(6, 9): sys.argv[1:] = self.allcondition[i] if i == 6: self.assertEqual(homework02.main(), 34) elif i == 7: self.assertEqual(homework02.main(), 45) elif i == 8: self.assertEqual(homework02.main(), 50) homework02.resetall()
def test_three_args(self): sys.argv[1:] = self.allcondition[10] self.assertEqual(homework02.main(), 54) homework02.resetall()
def test_two_args(self): sys.argv[1:] = self.allcondition[9] self.assertEqual(homework02.main(), 49) homework02.resetall()
def test_zero_arg(self): sys.argv[1:] = self.allcondition[5] self.assertEqual(homework02.main(), 28) homework02.resetall()