test.py: minor modifications