Update scheduler_test.py