Make the functions in test.py clearer.