Make tests work with py.test.

Don't comment - delete