Moving to py.test instead of nose