Adding a couple simple assertions to unit tests.