Cleaned up assertions in test cases.

Signed-off-by: Kevin Stone <kevinastone@gmail.com>