Move python unit tests