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