คู่มือเต็ม Python unittest: จากพื้นฐานสู่การประยุกต์ใช้

1. Python unittest คืออะไร?

ความสำคัญของการทดสอบหน่วย

2. วิธีการใช้ unittest เบื้องต้น

ตัวอย่างการทดสอบพื้นฐาน

import

# โค้ดที่ต้องทดสอบ
def add(a, b):
    return a + b

# คลาสทดสอบ
class TestAddFunction(unittest.TestCase):

    def test_add_integers(self):
        result = add(2, 3)
        self.assertEqual(result, 5)

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

การขยายการทดสอบ

def test_add_floats(self):
    result = add(2.5, 3.5)
    self.assertAlmostEqual(result, 6.0, places=2)

def test_add_strings(self):
    result = add("Hello, ", "World!")
    self.assertEqual(result, "Hello, World!")

Ad

3. วิธีการใช้ setUp() และ tearDown()

ตัวอย่าง setUp()

def setUp(self):
    self.temp_value = 42

ตัวอย่าง tearDown()

def tearDown(self):
    self.temp_value = None

4. การทดสอบความขึ้นอยู่ด้วยการใช้ Mock

ตัวอย่างการใช้ Mock

from unittest.mock import patch

class TestAddFunction(unittest.TestCase):

    @patch('my_module.time_consuming_function')
    def test_add_with_mock(self, mock_func):
        mock_func.return_value = 0
        result = add(2, 3)
        self.assertEqual(result, 5)

Ad

5. การจัดการข้อยกเว้นและการอ้างอิงแบบกำหนดเอง

การทดสอบการจัดการข้อยกเว้น

def test_divide_by_zero(self):
    with self.assertRaises(ZeroDivisionError):
        divide(1, 0)

การสร้างการอ้างอิงแบบกำหนดเอง

def assertIsPositive(self, value):
    self.assertTrue(value > 0, f'{value} is not positive')

6. ฟีเจอร์การค้นหาเทสต์ของ unittest

วิธีใช้การค้นหาเทสต์

python -m unittest discover

python -m unittest discover -s tests -p "test_*.py"

7. เคล็ดลับการเพิ่มประสิทธิภาพด้วย unittest

เพิ่มประสิทธิภาพไฟล์ I/O

from io import StringIO

class TestFileOperations(unittest.TestCase):

    def test_write_to_memory(self):
        output = StringIO()
        output.write('Hello, World!')
        self.assertEqual(output.getvalue(), 'Hello, World!')

ใช้ Mock

from unittest.mock import MagicMock

class TestApiCall(unittest.TestCase):

    def test_api_response(self):
        mock_api = MagicMock(return_value={'status': 'success'})
        response = mock_api()
        self.assertEqual(response['status'], 'success')

8. สรุปและขั้นตอนต่อไป

สรุปประเด็นสำคัญ

• วิธีการใช้พื้นฐาน: สืบทอดจาก unittest.TestCase และใช้เมธอด assertion เพื่อสร้างการทดสอบ
• setUp() / tearDown(): การจัดการกระบวนการทั่วไปก่อนและหลังการทดสอบช่วยเพิ่มการนำโค้ดกลับมาใช้ใหม่และความอ่านง่ายของโค้ด
• การใช้ mock: สามารถทดสอบฟังก์ชันโดยไม่ต้องพึ่งพาแหล่งข้อมูลภายนอก ทำให้ประสิทธิภาพของการทดสอบเพิ่มขึ้นอย่างมาก
• การค้นพบการทดสอบ (Test Discovery): ฟีเจอร์ที่สะดวกช่วยให้การจัดการการทดสอบในโครงการขนาดใหญ่ง่ายขึ้น
• เทคนิคการปรับปรุงประสิทธิภาพ: การประมวลผลในหน่วยความจำและการใช้ mock สามารถลดเวลาการรันของการทดสอบได้

ขั้นตอนต่อไป

Ad