Use assertEqual in wrapper testcase

The assertion failures are really hard to read when using just `assert`. Should I continue replacing them with the appropriate TestCase helpers like this?

Passes in py2.