Tests. Switch from nose into py.test