Update bfs_test.py