Switch to py.test for testing