Update test_functions.py