test_runner.py: add needed imports