How do I write unit tests in Python

Writing unit tests in Python is essential for ensuring that your code functions as expected. Unit tests are created using the built-in `unittest` framework or third-party libraries like `pytest`. They help you verify that individual parts of your code (units) behave correctly, which makes your codebase more reliable and easier to maintain.

Here’s a simple example of how to write a unit test using the `unittest` framework:

import unittest # Sample function to be tested def add(a, b): return a + b # Create a test case class TestMathFunctions(unittest.TestCase): def test_add(self): self.assertEqual(add(1, 2), 3) self.assertEqual(add(-1, 1), 0) self.assertEqual(add(-1, -1), -2) # Run the tests if __name__ == '__main__': unittest.main()

Python unit tests unittest pytest test-driven development