使用 unittest 进行单元测试

1import unittest
2
3def add_positive(a, b):
4    if a < 0 or b < 0:
5        return 0
6    return a + b
7
8class TestMathOperations(unittest.TestCase):
9    def test_addition(self):
10        result = add_positive(5, 10)
11        self.assertEqual(result, 15)
12
13    def test_negative_input(self):
14        result = add_positive(-5, 10)
15        self.assertEqual(result, 0)
16
17if __name__ == '__main__':
18    unittest.main()

1import unittest
2
3class TestStringUtilities(unittest.TestCase):
4    def test_string_formatting(self):
5        name = "python"
6        self.assertEqual(name.upper(), "PYTHON")
7        self.assertNotEqual(name, "PYTHON")
8
9    def test_membership(self):
10        fruits = ["apple", "banana", "cherry"]
11        self.assertIn("banana", fruits)
12        self.assertNotIn("orange", fruits)
13
14    def test_boolean_logic(self):
15        is_active = True
16        self.assertTrue(is_active)
17        self.assertFalse(5 < 2)
18
19if __name__ == '__main__':
20    unittest.main()

1import unittest
2
3def parse_percentage(value):
4    if not (0 <= value <= 100):
5        raise ValueError("Percentage must be between 0 and 100")
6    return value / 100
7
8class TestValidation(unittest.TestCase):
9    def test_value_error(self):
10        with self.assertRaises(ValueError):
11            parse_percentage(150)
12
13    def test_valid_percentage(self):
14        self.assertEqual(parse_percentage(50), 0.5)
15
16if __name__ == '__main__':
17    unittest.main()