Add more tests

That was intentional - the method I'm testing is `_assert_count_message`.