I have a simple test (test.sh), which checks whether the output returned by function.py and the expected output are the same:
python function.py\\ --input_fil