Better to use self.assertEqual