Converted all tests to py.test