Switched basic unittest to pytest style