MchcsCohortLibrary { public CohortDefinition age2Months() { AgeCohortDefinition age = new AgeCohortDefinition(); age.setName("Children with 2 months of age"); age.setMaxAge(2); age.setMaxAgeUnit(DurationUnit.MONTHS); age.addParameter(new Parameter("effectiveDate", "effective date", Date.class)); return age; }  CohortDefinition pcrWithinMonths(); CohortDefinition pcrInitialTest(); CohortDefinition age2Months(); CohortDefinition pcrInitialWithin2Months(); CohortDefinition ageBetween3And8Months(); CohortDefinition pcrInitialBetween3To8Months(); CohortDefinition serologyAntBodyTest(); CohortDefinition ageBetween9And12Months(); CohortDefinition serologyAntBodyTestBetween9And12Months(); CohortDefinition pcrBetween9And12MonthsAge(); CohortDefinition totalHeitestedBy12Months(); CohortDefinition detectedConfirmedStatus(); CohortDefinition pcrConfirmedPositive2Months(); CohortDefinition pcrConfirmedPositiveBetween3To8Months(); CohortDefinition pcrConfirmedPositiveBetween9To12Months(); CohortDefinition pcrTotalConfirmedPositive(); CohortDefinition ageAt6Months(); CohortDefinition exclusiveBreastFeeding(); CohortDefinition exclusiveReplacementFeeding(); CohortDefinition mixedFeeding(); CohortDefinition exclusiveBreastFeedingAtSixMonths(); CohortDefinition exclusiveReplacementFeedingAtSixMonths(); CohortDefinition mixedFeedingAtSixMonths(); CohortDefinition totalExposedAgedSixMoths(); CohortDefinition motherOnTreatmentAndBreastFeeding(); CohortDefinition motherOnTreatmentAndNotBreastFeeding(); CohortDefinition motherOnTreatmentAndNotBreastFeedingUnknown(); CohortDefinition totalBreastFeedingMotherOnTreatment(); CohortDefinition hivExposedInfants(); CohortDefinition hivExposedInfantsWithin2Months(); CohortDefinition hivExposedInfantsEligibleForCTX(); CohortDefinition hivExposedInfantsWithin2MonthsAndEligibleForCTX(); CohortDefinition scheduledVisitsInCWC();  }
@Test public void age2Months_shouldReturnInfantsAged2Months() throws Exception { CohortDefinition cd = mchcsCohortLibrary.age2Months(); context.addParameterValue("effectiveDate", TestUtils.date(2007, 7, 1)); EvaluatedCohort evaluated = Context.getService(CohortDefinitionService.class).evaluate(cd, context); ReportingTestUtils.assertCohortEquals(Arrays.asList(6), evaluated); }