Polish the tests: update assertions, comments, verify.