Use 'should' instead of 'assert'