Update dfs_test.py