Adding assert-like test