Refactor functest_utils.py