(minor) Updates to test_utils.py