Improve test_util.py