unittest দিয়ে Python কোড Test করা

# test_math.py
import unittest

def add(a, b):
    return a + b

class TestAdd(unittest.TestCase):
    def test_sums_two_numbers(self):
        self.assertEqual(add(2, 3), 5)

if __name__ == '__main__':
    unittest.main()

python -m unittest test_math.py
# .
# ----------------------------------------------------------------------
# Ran 1 test in 0.000s
# OK

def divide(a, b):
    if b == 0:
        raise ValueError('Cannot divide by zero')
    return a / b

class TestDivide(unittest.TestCase):
    def test_raises_on_zero(self):
        with self.assertRaises(ValueError):
            divide(10, 0)

class TestShoppingCart(unittest.TestCase):
    def setUp(self):
        self.cart = ShoppingCart()   # প্রতিটা test-এর আগে একটা নতুন cart

    def test_starts_empty(self):
        self.assertEqual(len(self.cart.items), 0)

    def test_add_item(self):
        self.cart.add('apple')
        self.assertIn('apple', self.cart.items)

python -m unittest discover
# test_*.py-তে মেলা প্রতিটা file স্বয়ংক্রিয়ভাবে খুঁজে চালায়