def test_findById(self): repo = PersonRepository() repo.add(Person(1, "Razvan", 123, "Cluj")) self.assertEqual(Person(1, "Razvan", 123, "Cluj"), repo.findById(1)) self.assertIsNone(repo.findById(3))