Examples with AS_ReadPosRankSumTest org.broadinstitute.hellbender.tools.walkers.annotator.allelespecific.AS_ReadPosRankSumTest used on opensource projects

Example 1 with AS_ReadPosRankSumTest

use of org.broadinstitute.hellbender.tools.walkers.annotator.allelespecific.AS_ReadPosRankSumTest in project gatk by broadinstitute.

the class ReadPosRankSumTestUnitTest method testReadPos_Raw.

@Test
public void testReadPos_Raw() {
    final AS_RankSumTest ann = new AS_ReadPosRankSumTest();
    final String key1 = GATKVCFConstants.AS_RAW_READ_POS_RANK_SUM_KEY;
    final String key2 = GATKVCFConstants.AS_READ_POS_RANK_SUM_KEY;
    final int[] startAlts = { 3, 4 };
    final int[] startRefs = { 1, 2 };
    final int readLength = 10;
    final List<GATKRead> refReads = Arrays.asList(makeRead(startRefs[0], 30), makeRead(startRefs[1], 30));
    final List<GATKRead> altReads = Arrays.asList(makeRead(startAlts[0], 30), makeRead(startAlts[1], 30));
    final ReadLikelihoods<Allele> likelihoods = AnnotationArtificialData.makeLikelihoods(sample1, refReads, altReads, -100.0, -100.0, REF, ALT);
    Assert.assertEquals(ann.getDescriptions().size(), 1);
    Assert.assertEquals(ann.getDescriptions().get(0).getID(), key1);
    Assert.assertEquals(ann.getKeyNames().size(), 1);
    Assert.assertEquals(ann.getKeyNames().get(0), key2);
    final ReferenceContext ref = null;
    //middle of the read
    final long position = 5L;
    final VariantContext vc = makeVC(position);
    final Map<String, Object> annotateRaw = ann.annotateRawData(ref, vc, likelihoods);
    final Map<String, Object> annotateNonRaw = ann.annotate(ref, vc, likelihoods);
    final String expected = startAlts[0] + ",1," + startAlts[1] + ",1" + AS_RankSumTest.PRINT_DELIM + startRefs[0] + ",1," + startRefs[1] + ",1";
    Assert.assertEquals(annotateRaw.get(key1), expected);
    Assert.assertEquals(annotateNonRaw.get(key1), expected);
    //past middle
    final long positionEnd = 8L;
    final VariantContext vcEnd = makeVC(positionEnd);
    //Note: past the middle of the read we compute the position from the end.
    final Map<String, Object> annotateEndRaw = ann.annotateRawData(ref, vcEnd, likelihoods);
    final Map<String, Object> annotateEndNonRaw = ann.annotate(ref, vcEnd, likelihoods);
    final String refS = (startRefs[0] + readLength - positionEnd - 1) + ",1," + (startRefs[1] + readLength - positionEnd - 1) + ",1";
    final String altS = (positionEnd - startAlts[1]) + ",1," + (positionEnd - startAlts[0]) + ",1";
    Assert.assertEquals(annotateEndRaw.get(key1), refS + AS_RankSumTest.PRINT_DELIM + altS);
    Assert.assertEquals(annotateEndNonRaw.get(key1), refS + AS_RankSumTest.PRINT_DELIM + altS);
    //past middle
    final long positionPastEnd = 20L;
    final VariantContext vcPastEnd = makeVC(positionPastEnd);
    //Note: past the end of the read, there's nothing
    final Map<String, Object> annotatePastEndRaw = ann.annotateRawData(ref, vcPastEnd, likelihoods);
    final Map<String, Object> annotatePastEndNonRaw = ann.annotate(ref, vcPastEnd, likelihoods);
    Assert.assertTrue(annotatePastEndRaw.isEmpty());
    Assert.assertTrue(annotatePastEndNonRaw.isEmpty());
}