Update index_test.py