Adds support for passing functions to `assert`