Adjust `assert` class to new `Screen#find` parameter types

[s1hofmann]

Short overview
Update assert.class to also accept the new parameter type used by Screen#find

Use case
Consistent API