ENH: py.test

Moving us toward py.test use

Closes #655.