Update test_psim.py