Switch to py.test as a testing framework