tests: migrate to py.test