updated test.py script for tests