One of my favorite aspects of Python is that it makes practicing TDD very easy. What makes it so frictionless is the doctest module. It allows you to write a test at the same time you define a function. No setup, no boilerplate, just write a function call and the expected output in the docstring. Here’s a quick example of a fibonacci function.
[sourcecode language=”python”]
def fib(n):
”’Return the nth fibonacci number.
>>> fib(0)
0
>>> fib(1)
1
>>> fib(2)
1
>>> fib(3)
2
>>> fib(4)
3
”’
if n == 0:
return 0
elif n == 1:
return 1
else:
return fib(n – 1) + fib(n – 2)
[/sourcecode]
If you want to run your doctests, just add the following three lines to the bottom of your module.
[sourcecode language=”python”]
if __name__ == ‘__main__’:
import doctest
doctest.testmod()
[/sourcecode]
Now you can run your module to run the doctests, like python fib.py.
So how well does this fit in with the TDD philosophy? Here’s the basic TDD practices.
- Think about what you want to test
- Write a small test
- Write just enough code to fail the test
- Run the test and watch it fail
- Write just enough code to pass the test
- Run the test and watch it pass (if it fails, go back to step 4)
- Go back to step 1 and repeat until done
And now a step-by-step breakdown of how to do this with doctests, in excruciating detail.
1. Define a new empty method
[sourcecode language=”python”]
def fib(n):
”’Return the nth fibonacci number.”’
pass
if __name__ == ‘__main__’:
import doctest
doctest.testmod()
[/sourcecode]
2. Write a doctest
[sourcecode language=”python”]
def fib(n):
”’Return the nth fibonacci number.
>>> fib(0)
0
”’
pass
[/sourcecode]
3. Run the module and watch the doctest fail
python fib.py
**********************************************************************
File "fib1.py", line 3, in __main__.fib
Failed example:
fib(0)
Expected:
0
Got nothing
**********************************************************************
1 items had failures:
1 of 1 in __main__.fib
***Test Failed*** 1 failures.4. Write just enough code to pass the failing doctest
[sourcecode language=”python”]
def fib(n):
”’Return the nth fibonacci number.
>>> fib(0)
0
”’
return 0
[/sourcecode]
5. Run the module and watch the doctest pass
python fib.py
6. Go back to step 2 and repeat
Now you can start filling in the rest of function, one test at time. In practice, you may not write code exactly like this, but the point is that doctests provide a really easy way to test your code as you write it.
Unit Tests
Ok, so doctests are great for simple tests. But what if your tests need to be a bit more complex? Maybe you need some external data, or mock objects. In that case, you’ll be better off with more traditional unit tests. But first, take a little time to see if you can decompose your code into a set of smaller functions that can be tested individually. I find that code that is easier to test is also easier to understand.
Running Tests
For running my tests, I use nose. I have a tests/ directory with a simple configuration file, nose.cfg
[nosetests] verbosity=3 with-doctest=1
Then in my Makefile, I add a test command so I can run make test.
test:
@nosetests --config=tests/nose.cfg tests PACKAGE1 PACKAGE2PACKAGE1 and PACKAGE2 are optional paths to your code. They could point to unit test packages and/or production code containing doctests.
And finally, if you’re looking for a continuous integration server, try Buildbot.
