Update test_simulation.py