gatk-3.8/public/java/test/org/broadinstitute/sting/utils/UtilsUnitTest.java

180 lines
7.2 KiB
Java
Raw Normal View History

/*
* Copyright (c) 2012 The Broad Institute
*
* Permission is hereby granted, free of charge, to any person
* obtaining a copy of this software and associated documentation
* files (the "Software"), to deal in the Software without
* restriction, including without limitation the rights to use,
* copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following
* conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
* OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
* HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR
* THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package org.broadinstitute.sting.utils;
import org.apache.commons.io.FileUtils;
import org.broadinstitute.sting.utils.io.IOUtils;
import org.testng.Assert;
import org.broadinstitute.sting.BaseTest;
import org.testng.annotations.Test;
import java.io.File;
import java.util.*;
/**
* Testing framework for general purpose utilities class.
*
* @author hanna
* @version 0.1
*/
public class UtilsUnitTest extends BaseTest {
@Test
public void testAppend() {
for ( int leftSize : Arrays.asList(0, 1, 2, 3) ) {
for ( final int rightSize : Arrays.asList(0, 1, 2) ) {
final List<Integer> left = new LinkedList<Integer>();
for ( int i = 0; i < leftSize; i++ ) left.add(i);
final List<Integer> total = new LinkedList<Integer>();
for ( int i = 0; i < leftSize + rightSize; i++ ) total.add(i);
if ( rightSize == 0 )
Assert.assertEquals(Utils.append(left), total);
if ( rightSize == 1 )
Assert.assertEquals(Utils.append(left, leftSize), total);
if ( rightSize == 2 )
Assert.assertEquals(Utils.append(left, leftSize, leftSize + 1), total);
}
}
}
@Test
public void testDupStringNoChars() {
String duped = Utils.dupString('a',0);
Assert.assertEquals(duped.length(), 0, "dupString did not produce zero-length string");
}
@Test
public void testDupStringOneChar() {
String duped = Utils.dupString('b',1);
Assert.assertEquals(duped.length(), 1, "dupString did not produce single character string");
Assert.assertEquals(duped.charAt(0), 'b', "dupString character was incorrect");
}
Expanded functionality for writing BAMs from HaplotypeCaller -- The new code includes a new mode to write out a BAM containing reads realigned to the called haplotypes from the HC, which can be easily visualized in IGV. -- Previous functionality maintained, with bug fixes -- Haplotype BAM writing code now lives in utils -- Created a base class that includes most of the functionality of writing reads realigned to haplotypes onto haplotypes. -- Created two subclasses, one that writes all haplotypes (previous functionality) and a CalledHaplotypeBAMWriter that will only write reads aligned to the actually called haplotypes -- Extended PerReadAlleleLikelihoodMap.getMostLikelyAllele to optionally restrict set of alleles to consider best -- Massive increase in unit tests in AlignmentUtils, along with several new powerful functions for manipulating cigars -- Fix bug in SWPairwiseAlignment that produces cigar elements with 0 size, and are now fixed with consolidateCigar in AlignmentUtils -- HaplotypeCaller now tracks the called haplotypes in the GenotypingEngine, and returns this information to the HC for use in visualization. -- Added extensive docs to HaplotypeCaller on how to use this capability -- BUGFIX -- don't modify the read bases in GATKSAMRecord in LikelihoodCalculationEngine in the HC -- Cleaned up SWPairwiseAlignment. Refactored out the big main and supplementary static methods. Added a unit test with a bug TODO to fix what seems to be an edge case bug in SW -- Integration test to make sure we can actually write a BAM for each mode. This test only ensures that the code runs and doesn't exception out. It doesn't actually enforce any MD5s -- HaplotypeBAMWriter also left aligns indels in the reads, as SW can return a random placement of a read against the haplotype. Calls leftAlign to make the alignments more clear, with unit test of real read to cover this case -- Writes out haplotypes for both all haplotype and called haplotype mode -- Haplotype writers now get the active region call, regardless of whether an actual call was made. Only emitting called haplotypes is moved down to CalledHaplotypeBAMWriter
2013-02-23 04:22:43 +08:00
@Test
public void testXor() {
Assert.assertEquals(Utils.xor(false, false), false, "xor F F failed");
Assert.assertEquals(Utils.xor(false, true), true, "xor F T failed");
Assert.assertEquals(Utils.xor(true, false), true, "xor T F failed");
Assert.assertEquals(Utils.xor(true, true), false, "xor T T failed");
}
@Test
public void testDupStringMultiChar() {
String duped = Utils.dupString('c',5);
Assert.assertEquals(duped.length(), 5, "dupString did not produce five character string");
Assert.assertEquals(duped,"ccccc","dupString string was incorrect");
}
@Test
public void testJoinMap() {
Map<String,Integer> map = new LinkedHashMap<String,Integer>();
map.put("one",1);
map.put("two",2);
String joined = Utils.joinMap("-",";",map);
Assert.assertTrue("one-1;two-2".equals(joined));
}
@Test
public void testJoinMapLargerSet() {
Map<String,Integer> map = new LinkedHashMap<String,Integer>();
map.put("one",1);
map.put("two",2);
map.put("three",1);
map.put("four",2);
map.put("five",1);
map.put("six",2);
String joined = Utils.joinMap("-",";",map);
Assert.assertTrue("one-1;two-2;three-1;four-2;five-1;six-2".equals(joined));
}
@Test
public void testEscapeExpressions() {
String[] expected, actual;
expected = new String[] {"one", "two", "three"};
actual = Utils.escapeExpressions("one two three");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" one two three");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions("one two three ");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" one two three ");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" one two three ");
Assert.assertEquals(actual, expected);
expected = new String[] {"one", "two", "three four", "five", "six"};
actual = Utils.escapeExpressions("one two 'three four' five six");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" one two 'three four' five six");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions("one two 'three four' five six ");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" one two 'three four' five six ");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" one two 'three four' five six ");
Assert.assertEquals(actual, expected);
expected = new String[] {"one two", "three", "four"};
actual = Utils.escapeExpressions("'one two' three four");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" 'one two' three four");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions("'one two' three four ");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" 'one two' three four ");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" 'one two' three four ");
Assert.assertEquals(actual, expected);
expected = new String[] {"one", "two", "three four"};
actual = Utils.escapeExpressions("one two 'three four'");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" one two 'three four'");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions("one two 'three four' ");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" one two 'three four' ");
Assert.assertEquals(actual, expected);
actual = Utils.escapeExpressions(" one two 'three four' ");
Assert.assertEquals(actual, expected);
}
@Test
public void testCalcMD5() throws Exception {
final File source = new File(publicTestDir + "exampleFASTA.fasta");
final String sourceMD5 = "36880691cf9e4178216f7b52e8d85fbe";
final byte[] sourceBytes = IOUtils.readFileIntoByteArray(source);
Assert.assertEquals(Utils.calcMD5(sourceBytes), sourceMD5);
final String sourceString = FileUtils.readFileToString(source);
Assert.assertEquals(Utils.calcMD5(sourceString), sourceMD5);
}
}