How can I automate unit testing in Python?

How can I automate unit testing in Python?

You can use frameworks like unittest or pytest to automate unit testing in Python. Would you like to see an example?

Yes, please provide an example using pytest.

Sure! First, install pytest using pip install pytest. Then, create a test file, for example, test_sample.py, and write your test functions prefixed with 'test_'. Here's a simple example:

# sample.py

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

# test_sample.py

from sample import add

def test_add():
    assert add(1, 2) == 3
    assert add(-1, 1) == 0

You can run the tests by executing pytest in your terminal.

What if I want to test a class method?

You can still use pytest for class methods. Here's an example:

# sample.py

class Calculator:
    def add(self, a, b):
        return a + b

# test_sample.py

from sample import Calculator

class TestCalculator:
    def test_add(self):
        calc = Calculator()
        assert calc.add(1, 2) == 3
        assert calc.add(-1, 1) == 0

This way, you can test methods within your classes.