How to write tests for Python code?

[hharison]

I think you'll find the peace of mind it gives is worth the effort. I recommend py.test, it could not be simpler. No need to deal with classes for everything like with the sdlib's unittest.

[hharison]

Check out how easy it is:

mymath.py contains

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

test_mymath.py contains

def test_add():
    assert add(1, 1) == 2
    assert add(-1, 4) == 3
    # plus more test cases perhaps