import unittest

def add(a, b):
    """
    این تابع دو مقدار رو با هم جمع می‌کنه، اما:
    - اگر هر کدوم از ورودی‌ها str باشه → خطا بده (TypeError)
    - اگر هر کدوم از ورودی‌ها None باشه → خطا بده
    - فقط اعداد int یا float رو قبول کنه
    """
    if not isinstance(a, (int, float)) or not isinstance(b, (int, float)):
        raise TypeError("ورودی‌ها باید عدد باشن!")
    if a is None or b is None:
        raise ValueError("ورودی نمی‌تونه None باشه!")
    
    return a + b


# ------------ تست‌ها --------------
class TestAddFunction(unittest.TestCase):
    
    # تست‌های موفق (Happy Path)
    def test_add_positive_numbers(self):
        self.assertEqual(add(2, 3), 
5)
    
    def test_add_float_numbers(self):
        self.assertEqual(add(2.5, 3.7), 6.2)
    
    def test_add_zero(self):
        self.assertEqual(add(10, 0), 10)

    # تست‌های خطادار (باید ارور بدن)
    def test_add_string_raises_type_error(self):
        with self.assertRaises(TypeError):
            add("2", 3)
    
    def test_add_none_raises_value_error(self):
        with self.assertRaises(ValueError):
            add(5, None)

   

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

چند تا نکته‌ی مهم که از همون اول باید جدی بگیری:

︎هر تست فقط یه چیز رو چک کنه.

︎اسم تست‌هاتو واضح انتخاب کن؛ که بعداً خودت هم بفهمی چی نوشتی.

︎تست‌هات وابسته به محیط و شانس و ساعت سیستم نباشن.

منتظر کوئیز ویژه این بخش باشید.