Update test framework to include tests for functions