Cleaning up Time py tests