Switch from nosetests to to py.test