switch to pytest