CLN: clean up test plotting

Corrects tests of the form

`self.assert_(x, y)` to `self.assertEqual(x, y)`. Also changes tests of the
form `self.assert_(x == y)` to `self.assertEqual(x, y)`.