def test_parse_divide(self):
args = parse_args(["-d", "60", "3"])
self.assertEqual(args.operand, [60.0, 3.0])
self.assertTrue(args.divide)
self.assertFalse(args.multiply)
self.assertFalse(args.subtract)
self.assertFalse(args.add)
def test_parse_multiply(self):
args = parse_args(["-m", "100", "2"])
self.assertEqual(args.operand, [100.0, 2.0])
self.assertTrue(args.multiply)
self.assertFalse(args.subtract)
self.assertFalse(args.add)
self.assertFalse(args.divide)
def test_parse_subtract(self):
args = parse_args(["-s", "7", "6.2"])
self.assertEqual(args.operand, [7.0, 6.2])
self.assertTrue(args.subtract)
self.assertFalse(args.add)
self.assertFalse(args.multiply)
self.assertFalse(args.divide)
def test_parse_add(self):
args = parse_args(["5", "10", "-a"])
self.assertEqual(args.operand, [5.0, 10.0])
self.assertTrue(args.add)
self.assertFalse(args.subtract)
self.assertFalse(args.multiply)
self.assertFalse(args.divide)
def test_parse_no_operator(self):
with mock.patch("basic_arithmetic.ArgumentParser.error") as mock_error:
parse_args(["60", "70"])
mock_error.assert_called()
def test_parse_three_operands(self):
with mock.patch("basic_arithmetic.ArgumentParser.error") as mock_error:
parse_args(["-d", "60", "70", "100"])
mock_error.assert_called()