Adds an ability to pass functions to `assert`