Update test.py

Modify to use print_function