Complete test_utils.py