def test_remove(self): repo = PersonRepository() repo.add(Person(1, "Razvan", 123, "Cluj")) repo.add(Person(2, "Razvan", 321, "Zalau")) self.assertEqual(repo.remove(2), Person(2, "Razvan", 321, "Zalau"))