test.py with simple while loop