Switch from nose to pytest