Use self.assertEqual instead of assert

Change-Id: Icae26d907eec2c4ab9f430c69491a96bd549c45a