use pytest for the tests