Add comments, make comparison less strict

Using `assertEqual` should allow the intent of the test to be validated, without
worrying about implementation issues like "does the mock library swap out
objects under the hood?"