Add tests for type_check callbacks
Rename "type_check_inputs" to "type_check"

This way the function name matches the callback name, like execute.
Add more extensive callback tests