joel/largeRCRF-Java
1
0
Fork 0
Code Issues 3 Pull requests Releases Wiki Activity

Add test for CompetingRiskFunctions

Browse source
This commit is contained in:
Joel Therrien 2018-08-31 22:32:54 -07:00
parent 8333579a1f
commit 2fb80df5a5
1 changed files with 47 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
src/test/java/ca/joeltherrien/randomforest/competingrisk/TestCompetingRiskFunctions.java Normal file
View file

@ -0,0 +1,47 @@
package ca.joeltherrien.randomforest.competingrisk;
import ca.joeltherrien.randomforest.TestUtils;
import ca.joeltherrien.randomforest.responses.competingrisk.CompetingRiskFunctions;
import ca.joeltherrien.randomforest.utils.MathFunction;
import ca.joeltherrien.randomforest.utils.Point;
import ca.joeltherrien.randomforest.utils.Utils;
import org.junit.jupiter.api.Test;
import java.util.Collections;
public class TestCompetingRiskFunctions {
@Test
public void testCalculateEventSpecificMortality(){
final MathFunction cif1 = new MathFunction(
Utils.easyList(
new Point(1.0, 0.3),
new Point(1.5, 0.7),
new Point(2.0, 0.8)
), new Point(0.0 ,0.0)
);
// not being used
final MathFunction chf1 = new MathFunction(Collections.emptyList());
// not being used
final MathFunction km = new MathFunction(Collections.emptyList());
final CompetingRiskFunctions functions = CompetingRiskFunctions.builder()
.causeSpecificHazards(Collections.singletonList(chf1))
.cumulativeIncidenceCurves(Collections.singletonList(cif1))
.survivalCurve(km)
.build();
final double mortality1_5 = functions.calculateEventSpecificMortality(1, 1.5);
final double mortality3 = functions.calculateEventSpecificMortality(1, 3.0);
final double mortality4 = functions.calculateEventSpecificMortality(1, 4.0);
TestUtils.closeEnough(0.15, mortality1_5, 0.001);
TestUtils.closeEnough(1.3, mortality3, 0.001);
TestUtils.closeEnough(2.1, mortality4, 0.001);
}
}