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()
