Minor Python 3 sorted tweak
Use unittest.assertEquals etc, not assert statements