changed the python tests to py.test