Пример #1
0
  @Test
  public final void testApplyChanges() throws InvalidIdentifierException {
    final RiskModel riskModel = createRiskModel("Thoracic 30-day Mortality");
    final EditRiskModel editRiskModel = EditRiskModel.fromRiskModel(riskModel, fModelService);
    final ImmutableList<EditModelTerm> newTerms =
        ImmutableList.of(
            EditModelTerm.forConstant(0.23f),
            EditModelTerm.forVariable("dnr", 12.3f),
            EditModelTerm.forRule(SampleModels.ageAndFsRule().getDisplayName(), 0.5f));
    final HashSet<ModelTerm> expectedTerms = new HashSet<>();
    for (final EditModelTerm newTerm : newTerms) {
      expectedTerms.add(newTerm.build(fModelService));
    }

    editRiskModel.setModelName("NewModelName");
    editRiskModel.getTerms().clear();
    editRiskModel.getTerms().addAll(newTerms);

    assertEquals("NewModelName", editRiskModel.getModelName());
    assertEquals(DisplayNameConditions.DISPLAY_NAME_MAX, editRiskModel.getMaxDisplayNameLength());

    editRiskModel.applyChanges(riskModel, fModelService);

    assertEquals("NewModelName", riskModel.getDisplayName());
    assertEquals(expectedTerms, riskModel.getTerms());
  }
Пример #2
0
  @Test
  public final void testCalculateDependentModels() {
    // Setup
    final BooleanVariable var = SampleModels.dnrVariable();
    // Use an EditBooleanVar to test since it is a very basic
    // implementation of EditBaseVar.
    final EditBaseVar ev = new EditBooleanVar(var, fModelService);

    // Verification
    assertEquals(ImmutableSet.of(fModelService.getThoracicModel()), ev.getDependentModels());
  }
Пример #3
0
  @Test
  public final void testCardiacAge() {
    final DiscreteNumericalVariable var = SampleModels.cardiacAgeVariable();
    final Patient patient = SampleCalculations.dummyPatientWithVitals(1);
    final Collection<DiscreteNumericalVariable> vars = ImmutableSet.of(var);
    // Use VariableEntry to generate the expected dynamic values.
    final VariableEntry expected =
        new VariableEntry(vars)
            .putDynamicValue(
                VariableEntry.makeNumericalInputName(var.getKey()),
                String.valueOf(patient.getAge()));

    verifyRetrievedValue(vars, patient, expected.getDynamicValues());
  }
Пример #4
0
 /**
  * This method returns a DiscreteNumericalVariable with the specified key and retriever, but with
  * dummy information for the other fields in the variable.
  *
  * @param key the key to use for the variable
  * @param retriever the ValueRetriever for this variable
  * @return the constructed DiscreteNumericalVariable
  */
 private DiscreteNumericalVariable makeDiscreteNumerical(
     final String key, final ValueRetriever retriever) {
   final Category wbcWnl = new Category(new MultiSelectOption("WNL"), 11.0f, true);
   final Category wbcHigh =
       new Category(new MultiSelectOption(">11.0"), Float.POSITIVE_INFINITY, false);
   final DiscreteNumericalVariable var =
       new DiscreteNumericalVariable(
           "Generic Display Name",
           SampleModels.labVariableGroup(),
           ImmutableSet.of(wbcWnl, wbcHigh),
           key);
   var.setValidRange(new NumericalRange(2.0f, true, 50.0f, true));
   var.setUnits("generic units");
   var.setRetriever(retriever);
   return var;
 }
Пример #5
0
  /**
   * Constructs a sample RiskModel for our testing.
   *
   * @param name
   * @return
   */
  private RiskModel createRiskModel(String name) {
    final ProcedureVariable procedureVar = SampleModels.procedureVariable();
    final NumericalVariable ageVar = SampleModels.ageVariable();
    final DiscreteNumericalVariable wbcVar = SampleModels.wbcVariable();
    final MultiSelectVariable fsVar = SampleModels.functionalStatusVariable();
    // Note: intentionally including this edge-case variable to test ordering.
    final BooleanVariable wbcIsNormalVar = SampleModels.wbcIsNormalVariable();
    final Set<DerivedTerm> derivedTerms = new HashSet<DerivedTerm>();
    derivedTerms.add(new DerivedTerm(6.0f, fTestRule));

    return SampleModels.makeSampleRiskModel(
        name, derivedTerms, procedureVar, wbcIsNormalVar, ageVar, wbcVar, fsVar);
  }
Пример #6
0
/** Tests the {@link EditRiskModel} class. */
public class EditRiskModelTest {
  private final MockModelService fModelService = new MockModelService();

  private final Rule fTestRule = SampleModels.ageAndFsRule();

  /**
   * Constructs a sample RiskModel for our testing.
   *
   * @param name
   * @return
   */
  private RiskModel createRiskModel(String name) {
    final ProcedureVariable procedureVar = SampleModels.procedureVariable();
    final NumericalVariable ageVar = SampleModels.ageVariable();
    final DiscreteNumericalVariable wbcVar = SampleModels.wbcVariable();
    final MultiSelectVariable fsVar = SampleModels.functionalStatusVariable();
    // Note: intentionally including this edge-case variable to test ordering.
    final BooleanVariable wbcIsNormalVar = SampleModels.wbcIsNormalVariable();
    final Set<DerivedTerm> derivedTerms = new HashSet<DerivedTerm>();
    derivedTerms.add(new DerivedTerm(6.0f, fTestRule));

    return SampleModels.makeSampleRiskModel(
        name, derivedTerms, procedureVar, wbcIsNormalVar, ageVar, wbcVar, fsVar);
  }

  @Test
  public final void testApplyChanges() throws InvalidIdentifierException {
    final RiskModel riskModel = createRiskModel("Thoracic 30-day Mortality");
    final EditRiskModel editRiskModel = EditRiskModel.fromRiskModel(riskModel, fModelService);
    final ImmutableList<EditModelTerm> newTerms =
        ImmutableList.of(
            EditModelTerm.forConstant(0.23f),
            EditModelTerm.forVariable("dnr", 12.3f),
            EditModelTerm.forRule(SampleModels.ageAndFsRule().getDisplayName(), 0.5f));
    final HashSet<ModelTerm> expectedTerms = new HashSet<>();
    for (final EditModelTerm newTerm : newTerms) {
      expectedTerms.add(newTerm.build(fModelService));
    }

    editRiskModel.setModelName("NewModelName");
    editRiskModel.getTerms().clear();
    editRiskModel.getTerms().addAll(newTerms);

    assertEquals("NewModelName", editRiskModel.getModelName());
    assertEquals(DisplayNameConditions.DISPLAY_NAME_MAX, editRiskModel.getMaxDisplayNameLength());

    editRiskModel.applyChanges(riskModel, fModelService);

    assertEquals("NewModelName", riskModel.getDisplayName());
    assertEquals(expectedTerms, riskModel.getTerms());
  }

  @Test
  public final void testGetTermSummaries() {
    /* Setup */
    final RiskModel riskModel = createRiskModel("Thoracic 30-day Mortality");
    final float constant = 0.34f;
    riskModel.setConstantTerm(new ConstantTerm(constant));
    final EditRiskModel editRiskModel = EditRiskModel.fromRiskModel(riskModel, fModelService);
    // Constant first, then the Rules and then the rest
    final ImmutableList<ModelTermSummary> expectedSummaries =
        ImmutableList.of(
            new ModelTermSummary("Constant", "", constant),
            new ModelTermSummary(fTestRule.getDisplayName(), "Rule", 6.0f),
            new ModelTermSummary("Age", "Numerical", 2.0f),
            new ModelTermSummary("Functional Status = Independent", "Multi-Select", 3.0f),
            new ModelTermSummary("Functional Status = Partially dependent", "Multi-Select", 3.0f),
            new ModelTermSummary("Functional Status = Totally dependent", "Multi-Select", 3.0f),
            new ModelTermSummary("Procedure", "Procedure (RVU Multiplier)", 1.0f),
            new ModelTermSummary("White Blood Count = WNL", "Discrete Numerical", 4.0f),
            new ModelTermSummary("White Blood Count = >11.0", "Discrete Numerical", 4.0f),
            new ModelTermSummary("White Blood Count is Normal", "Checkbox", 5.0f));

    /* Behavior */
    ImmutableList<ModelTermSummary> actualSummaries =
        editRiskModel.makeTermSummaries(fModelService);

    /* Verification */
    assertEquals(expectedSummaries, actualSummaries);
  }
}