Updated test.py
Improve replace_world method.