zzh
/
gatk-3.8
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Test multiple bases and intervals

This commit is contained in:
Joel Thibault 2012-11-19 14:44:00 -05:00
parent c08b782743
commit e8defcb20d
1 changed files with 45 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
public/java/test/org/broadinstitute/sting/gatk/traversals/TraverseActiveRegionsTest.java
View File

@ -83,49 +83,74 @@ public class TraverseActiveRegionsTest extends BaseTest {
} }
@Test @Test
public void testAllIntervalsSeen() throws Exception { public void testAllBasesSeenSuite() {
List<GenomeLoc> intervals = new ArrayList<GenomeLoc>(); List<GenomeLoc> intervals = new ArrayList<GenomeLoc>();
List<GenomeLoc> activeIntervals = new ArrayList<GenomeLoc>(); List<GenomeLoc> activeIntervals = new ArrayList<GenomeLoc>();
GenomeLoc interval = genomeLocParser.createGenomeLoc("1", 1, 1); GenomeLoc interval = genomeLocParser.createGenomeLoc("1", 1, 1);
intervals.add(interval); intervals.add(interval);
testAllBasesSeen(intervals);
LocusShardDataProvider dataProvider = createDataProvider(intervals); interval = genomeLocParser.createGenomeLoc("1", 10, 20);
intervals.add(interval);
testAllBasesSeen(intervals);
}
t.traverse(walker, dataProvider, 0); public void testAllBasesSeen(List<GenomeLoc> intervals) {
activeIntervals.addAll(walker.isActiveCalls); List<GenomeLoc> activeIntervals = new ArrayList<GenomeLoc>();
for (LocusShardDataProvider dataProvider : createDataProviders(intervals)) {
t.traverse(walker, dataProvider, 0);
activeIntervals.addAll(walker.isActiveCalls);
}
boolean allGenomeLocsSeen = true; boolean allBasesSeen = true;
for (GenomeLoc loc : intervals) { for (GenomeLoc base : toBases(intervals)) {
boolean thisGenomeLocSeen = false; boolean thisBaseSeen = false;
for (GenomeLoc activeLoc : activeIntervals) { for (GenomeLoc activeLoc : activeIntervals) {
if (loc.equals(activeLoc)) { if (base.equals(activeLoc)) {
thisGenomeLocSeen = true; thisBaseSeen = true;
break; break;
} }
} }
if (!thisGenomeLocSeen) { if (!thisBaseSeen) {
allGenomeLocsSeen = false; allBasesSeen = false;
break; break;
} }
} }
Assert.assertTrue(allGenomeLocsSeen, "Some intervals missing from activity profile"); Assert.assertTrue(allBasesSeen, "Some intervals missing from activity profile");
} }
private LocusShardDataProvider createDataProvider(List<GenomeLoc> intervals) { private List<GenomeLoc> toBases(List<GenomeLoc> intervals) {
List<GenomeLoc> bases = new ArrayList<GenomeLoc>();
for (GenomeLoc interval : intervals) {
if (interval.size() == 1)
bases.add(interval);
else {
for (int location = interval.getStart(); location <= interval.getStop(); location++) {
bases.add(genomeLocParser.createGenomeLoc(interval.getContig(), location, location));
}
}
}
return bases;
}
private List<LocusShardDataProvider> createDataProviders(List<GenomeLoc> intervals) {
walker = new DummyActiveRegionWalker(); walker = new DummyActiveRegionWalker();
StingSAMIterator iterator = ArtificialSAMUtils.createReadIterator(new ArrayList<SAMRecord>());
Shard shard = new MockLocusShard(genomeLocParser, intervals);
WindowMaker windowMaker = new WindowMaker(shard, genomeLocParser,iterator,shard.getGenomeLocs());
WindowMaker.WindowMakerIterator window = windowMaker.next();
GenomeAnalysisEngine engine = new GenomeAnalysisEngine(); GenomeAnalysisEngine engine = new GenomeAnalysisEngine();
//engine.setReferenceDataSource(reference);
engine.setGenomeLocParser(genomeLocParser); engine.setGenomeLocParser(genomeLocParser);
t.initialize(engine); t.initialize(engine);
return new LocusShardDataProvider(shard, null, genomeLocParser, window.getLocus(), window, reference, new ArrayList<ReferenceOrderedDataSource>()); StingSAMIterator iterator = ArtificialSAMUtils.createReadIterator(new ArrayList<SAMRecord>());
Shard shard = new MockLocusShard(genomeLocParser, intervals);
List<LocusShardDataProvider> providers = new ArrayList<LocusShardDataProvider>();
WindowMaker windowMaker = new WindowMaker(shard,genomeLocParser,iterator,shard.getGenomeLocs());
for (WindowMaker.WindowMakerIterator window : windowMaker) {
providers.add(new LocusShardDataProvider(shard, null, genomeLocParser, window.getLocus(), window, reference, new ArrayList<ReferenceOrderedDataSource>()));
}
return providers;
} }
} }