We can actually use `assertContains` and `assertNotContains` to simplify things here. I'm making the change and committing this.