Exemplo n.º 1
0
  @Test
  public void testStrataNamesAndIdsStrataCriterions() {
    DichotomousCriterion criterion1 = new DichotomousCriterion();
    criterion1.setId(1);
    criterion1.setName("criterion1");
    criterion1.setOption1("option1");
    criterion1.setOption2("option2");
    try {
      DichotomousConstraint d1 = new DichotomousConstraint(Arrays.asList(new String[] {"option1"}));
      d1.setId(1);
      criterion1.addStrata(d1);
      DichotomousConstraint d2 = new DichotomousConstraint(Arrays.asList(new String[] {"option2"}));
      d2.setId(2);
      criterion1.addStrata(d2);
    } catch (ContraintViolatedException e) {
      fail();
    }
    DichotomousCriterion criterion2 = new DichotomousCriterion();
    criterion2.setId(2);
    criterion2.setName("criterion2");
    criterion2.setOption1("option1");
    criterion2.setOption2("option2");
    try {
      DichotomousConstraint d1 = new DichotomousConstraint(Arrays.asList(new String[] {"option1"}));
      d1.setId(1);
      criterion2.addStrata(d1);
      DichotomousConstraint d2 = new DichotomousConstraint(Arrays.asList(new String[] {"option2"}));
      d2.setId(2);
      criterion2.addStrata(d2);
    } catch (ContraintViolatedException e) {
      fail();
    }
    validTrial.addCriterion(criterion1);
    validTrial.addCriterion(criterion2);

    Pair<List<String>, List<String>> pair = validTrial.getAllStrataIdsAndNames();

    assertEquals(4, pair.first().size());
    assertEquals(4, pair.last().size());

    Collections.sort(pair.first());
    assertEquals("1_1;2_1;", pair.first().get(0));
    assertEquals("1_1;2_2;", pair.first().get(1));
    assertEquals("1_2;2_1;", pair.first().get(2));
    assertEquals("1_2;2_2;", pair.first().get(3));

    Collections.sort(pair.last());
    assertEquals("criterion1_option1;criterion2_option1;", pair.last().get(0));
    assertEquals("criterion1_option1;criterion2_option2;", pair.last().get(1));
    assertEquals("criterion1_option2;criterion2_option1;", pair.last().get(2));
    assertEquals("criterion1_option2;criterion2_option2;", pair.last().get(3));
  }
Exemplo n.º 2
0
  @Test
  public void testChartSubGroups() {
    authenticatAsPrincipalInvestigator();
    validTrial.setStartDate(new GregorianCalendar(2009, 0, 1));
    validTrial.setEndDate(new GregorianCalendar(2010, 11, 1));
    int blocksize = 4;
    int randomizations = 240;
    TreatmentArm arm1 = new TreatmentArm();
    arm1.setPlannedSubjects(randomizations / 2);
    arm1.setName("arm1");
    arm1.setTrial(validTrial);
    TreatmentArm arm2 = new TreatmentArm();
    arm2.setPlannedSubjects(randomizations / 2);
    arm2.setName("arm2");
    arm2.setTrial(validTrial);
    List<TreatmentArm> arms = new ArrayList<TreatmentArm>();
    arms.add(arm1);
    arms.add(arm2);

    validTrial.setTreatmentArms(arms);
    DichotomousCriterion cr = new DichotomousCriterion();
    cr.setOption1("1");
    cr.setOption2("2");
    DichotomousCriterion cr1 = new DichotomousCriterion();
    cr1.setOption1("1");
    cr1.setOption2("2");
    DichotomousCriterion cr2 = new DichotomousCriterion();
    cr2.setOption1("1");
    cr2.setOption2("2");
    try {

      cr.addStrata(new DichotomousConstraint(Arrays.asList(new String[] {"1"})));

      cr.addStrata(new DichotomousConstraint(Arrays.asList(new String[] {"2"})));

      cr1.addStrata(new DichotomousConstraint(Arrays.asList(new String[] {"1"})));
      cr1.addStrata(new DichotomousConstraint(Arrays.asList(new String[] {"2"})));
      cr2.addStrata(new DichotomousConstraint(Arrays.asList(new String[] {"1"})));
      cr2.addStrata(new DichotomousConstraint(Arrays.asList(new String[] {"2"})));

      validTrial.addCriterion(cr);
      validTrial.addCriterion(cr1);
      validTrial.addCriterion(cr2);

    } catch (ContraintViolatedException e) {
      BoxedException.throwBoxed(e);
    }

    BlockRandomizationConfig config = new BlockRandomizationConfig();
    config.setMaximum(blocksize);
    config.setMinimum(blocksize);
    validTrial.setRandomizationConfiguration(config);
    trialService.create(validTrial);
    assertTrue(validTrial.getId() > 0);
    assertEquals(2, validTrial.getTreatmentArms().size());
    authenticatAsInvestigator();
    for (int i = 0; i < randomizations; i++) {
      TrialSubject subject = new TrialSubject();
      subject.setIdentification("identification" + i);
      subject.setTrialSite(validTrial.getLeadingSite());
      SubjectProperty<Serializable> subprob =
          new SubjectProperty<Serializable>(validTrial.getCriteria().get(0));
      SubjectProperty<Serializable> subprob1 =
          new SubjectProperty<Serializable>(validTrial.getCriteria().get(1));
      SubjectProperty<Serializable> subprob2 =
          new SubjectProperty<Serializable>(validTrial.getCriteria().get(2));
      try {
        if ((new Random()).nextInt(2) == 0) {
          subprob.setValue(cr.getOption1());
        } else {
          subprob.setValue(cr.getOption2());
        }

        if ((new Random()).nextInt(2) == 0) {
          subprob1.setValue(cr1.getOption1());
        } else {
          subprob1.setValue(cr1.getOption2());
        }
        if ((new Random()).nextInt(2) == 0) {
          subprob2.setValue(cr2.getOption1());
        } else {
          subprob2.setValue(cr2.getOption2());
        }

      } catch (ContraintViolatedException e) {
        BoxedException.throwBoxed(e);
      }

      Set<SubjectProperty<?>> proberties = new HashSet<SubjectProperty<?>>();
      proberties.add(subprob);
      proberties.add(subprob1);
      proberties.add(subprob2);
      subject.setProperties(proberties);
      trialService.randomize(validTrial, subject);
      subject.setCreatedAt(new GregorianCalendar(2009 + (i >= 120 ? 1 : 0), i % 12, 1));
      sessionFactory.getCurrentSession().update(subject);
    }
    ChartData data = chartsService.generateRecruitmentChartFactors(validTrial);
    assertEquals(8, data.getData().size());
    assertEquals(8, data.getData().get(0).length);
    assertEquals(8, data.getXLabels().size());
  }