diff --git a/src/test/java/org/broadinstitute/hellbender/tools/walkers/mutect/SomaticGenotypingEngineUnitTest.java b/src/test/java/org/broadinstitute/hellbender/tools/walkers/mutect/SomaticGenotypingEngineUnitTest.java index 09b90610f44..c99fe8be850 100644 --- a/src/test/java/org/broadinstitute/hellbender/tools/walkers/mutect/SomaticGenotypingEngineUnitTest.java +++ b/src/test/java/org/broadinstitute/hellbender/tools/walkers/mutect/SomaticGenotypingEngineUnitTest.java @@ -10,6 +10,7 @@ import org.testng.annotations.DataProvider; import org.testng.annotations.Test; +import javax.ws.rs.core.Variant; import java.io.IOException; import java.util.ArrayList; import java.util.Arrays; @@ -88,4 +89,18 @@ public void testGetGermlineAltAlleleFrequenciesWithMissingAF(final VariantContex final double[] result = SomaticGenotypingEngine.getGermlineAltAlleleFrequencies(vc.getAlleles(), List.of(vc), DEFAULT_AF); Assert.assertEquals(result, new double[] {DEFAULT_AF}, 1.0e-10); } + + // test getting alt allele frequencies when each alt has its own VCF line and its own VariantContext + @Test + public void testGetGermlineAltAlleleFrequenciesFromSplitAllelesFormat() { + final double af1 = 0.1; + final double af2 = 0.2; + final VariantContext vc1 = new VariantContextBuilder("SOURCE", "1", 1, 1, Arrays.asList(Allele.REF_A, Allele.ALT_C)) + .attribute(VCFConstants.ALLELE_FREQUENCY_KEY, af1).make(); + final VariantContext vc2 = new VariantContextBuilder("SOURCE", "1", 1, 1, Arrays.asList(Allele.REF_A, Allele.ALT_T)) + .attribute(VCFConstants.ALLELE_FREQUENCY_KEY, af2).make(); + final List alleles = List.of(Allele.REF_A, Allele.ALT_C, Allele.ALT_T); + final double[] result = SomaticGenotypingEngine.getGermlineAltAlleleFrequencies(alleles, List.of(vc1, vc2), DEFAULT_AF); + Assert.assertEquals(result, new double[] {af1, af2}, 1.0e-10); + } } \ No newline at end of file