Update test_scenarios.py