### Question:
TreatmentCategorizationPresentation extends PresentationModel<TreatmentCategorization> { public StudyListPresentation getCategorizedStudyList(final Category category) { return d_studyListPresentations.get(category); } TreatmentCategorizationPresentation(final TreatmentCategorization bean, final Domain domain); StudyListPresentation getCategorizedStudyList(final Category category); DrugPresentation getDrugPresentation();  }### Answer:
@Test public void testIncludedStudies() { final Study studyBennie = ExampleData.buildStudyBennie(); final Study studyChouinard = ExampleData.buildStudyChouinard(); final Study studyFava2002 = ExampleData.buildStudyFava2002(); final Category foo = new Category(d_tc, "foo"); final Category bar = new Category(d_tc, "bar"); final Category baz = new Category(d_tc, "baz"); d_domain.getAdverseEvents().add(ExampleData.buildAdverseEventSexualDysfunction()); d_domain.getStudies().add(studyFava2002); d_wpm.getCategories().add(foo); d_wpm.getCategories().add(bar); d_wpm.getCategories().add(baz); d_wpm.getModelForFixedDose().setValue(d_wpm.getFixedRangeNode()); d_wpm.addDefaultRangeEdge(d_wpm.getFixedRangeNode()); Pair<RangeEdge> splits = d_wpm.splitRange((RangeEdge) d_wpm.getOutEdges(d_wpm.getFixedRangeNode()).get(0), 21.0, false); d_wpm.getModelForEdge(splits.getFirst()).setValue(new LeafNode(foo)); d_wpm.getModelForEdge(splits.getSecond()).setValue(new LeafNode(bar)); d_pm = new TreatmentCategorizationPresentation(d_tc, d_domain); final ObservableList<Study> fooStudies = d_pm.getCategorizedStudyList(foo).getIncludedStudies(); final ObservableList<Study> barStudies = d_pm.getCategorizedStudyList(bar).getIncludedStudies(); final ObservableList<Study> bazStudies = d_pm.getCategorizedStudyList(baz).getIncludedStudies(); assertTrue(fooStudies.contains(studyBennie)); assertFalse(fooStudies.contains(studyChouinard)); assertFalse(fooStudies.contains(studyFava2002)); assertTrue(barStudies.contains(studyChouinard)); assertTrue(barStudies.contains(studyFava2002)); assertTrue(bazStudies.isEmpty()); }