Move to py.test.  Closes #9.