Might be worth adding to these tests the other exceptions that are raised within the code (which if I'm correct are only the `AssertionError` ones).