def test_findActivityPersons(self): repo = PersonsActivitiesRepo() cont = PersonActivityController(repo, UndoRedoController()) cont.addPersonActivity(1, 1) cont.addPersonActivity(1, 2) cont.addPersonActivity(1, 3) cont.addPersonActivity(2, 1) cont.addPersonActivity(2, 2) cont.addPersonActivity(3, 1) cont.addPersonActivity(4, 2) self.assertEqual(cont.findActivityPersons(1), repo.getActivityPersons(1))