Reorder imports

According to PEP8, `mock` should be with the rest of the third party libraries.