updated test.py
updated test.py