Updated test_add_func.py