Move to py.test for running tests