changing my test into pytest