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

Add simple main function to unit test so we can run the nano scheduler test from the command line

This commit is contained in:
Mark DePristo 2012-08-27 09:24:16 -04:00
parent 2d1ea7124b
commit e5b1f1c7f4
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
public/java/test/org/broadinstitute/sting/utils/nanoScheduler/NanoSchedulerUnitTest.java
View File

@ -18,11 +18,11 @@ import java.util.*;
public class NanoSchedulerUnitTest extends BaseTest { public class NanoSchedulerUnitTest extends BaseTest {
public static final int NANO_SCHEDULE_MAX_RUNTIME = 60000; public static final int NANO_SCHEDULE_MAX_RUNTIME = 60000;
private class Map2x implements MapFunction<Integer, Integer> { private static class Map2x implements MapFunction<Integer, Integer> {
@Override public Integer apply(Integer input) { return input * 2; } @Override public Integer apply(Integer input) { return input * 2; }
} }
private class ReduceSum implements ReduceFunction<Integer, Integer> { private static class ReduceSum implements ReduceFunction<Integer, Integer> {
int prevOne = Integer.MIN_VALUE; int prevOne = Integer.MIN_VALUE;
@Override public Integer apply(Integer one, Integer sum) { @Override public Integer apply(Integer one, Integer sum) {
@ -38,7 +38,7 @@ public class NanoSchedulerUnitTest extends BaseTest {
return sum; return sum;
} }
private class NanoSchedulerBasicTest extends TestDataProvider { private static class NanoSchedulerBasicTest extends TestDataProvider {
final int bufferSize, mapGroupSize, nThreads, start, end, expectedResult; final int bufferSize, mapGroupSize, nThreads, start, end, expectedResult;
public NanoSchedulerBasicTest(final int bufferSize, final int mapGroupSize, final int nThreads, final int start, final int end) { public NanoSchedulerBasicTest(final int bufferSize, final int mapGroupSize, final int nThreads, final int start, final int end) {
@ -146,4 +146,13 @@ public class NanoSchedulerUnitTest extends BaseTest {
nanoScheduler.shutdown(); nanoScheduler.shutdown();
nanoScheduler.execute(exampleTest.makeReader(), exampleTest.makeMap(), exampleTest.initReduce(), exampleTest.makeReduce()); nanoScheduler.execute(exampleTest.makeReader(), exampleTest.makeMap(), exampleTest.initReduce(), exampleTest.makeReduce());
} }
public static void main(String [ ] args) {
final NanoSchedulerBasicTest test = new NanoSchedulerBasicTest(1000, 100, Integer.valueOf(args[0]), 0, Integer.valueOf(args[1]));
final NanoScheduler<Integer, Integer, Integer> nanoScheduler =
new NanoScheduler<Integer, Integer, Integer>(test.bufferSize, test.mapGroupSize, test.nThreads);
final Integer sum = nanoScheduler.execute(test.makeReader(), test.makeMap(), test.initReduce(), test.makeReduce());
System.out.printf("Sum = %d, expected =%d%n", sum, test.expectedResult);
}
} }