Update dz_test.py