refactor tests to use only py.test