Update test functions.py