Move test file under py/