Removed an unused variable.
Switched to py.test.