public static void stall_till_cloudsize(String[] args, int x) {
if (!_stall_called_before) {
H2O.main(args);
H2O.registerRestApis(System.getProperty("user.dir"));
_stall_called_before = true;
}
H2O.waitForCloudSize(x, 30000);
_initial_keycnt = H2O.store_size();
}
public static void main(String[] args) {
try {
H2O.main(args);
TestUtil.stall_till_cloudsize(3);
List<Class> tests = JUnitRunner.all();
Result r = org.junit.runner.JUnitCore.runClasses(tests.toArray(new Class[0]));
if( r.getFailureCount() == 0 ) {
System.out.println("Successfully ran the following tests in " + (r.getRunTime() / 1000) + "s");
for( Class c : tests )
System.out.println(c.getName());
} else {
for( Failure f : r.getFailures() ) {
System.err.println(f.getDescription());
if( f.getException() != null )
f.getException().printStackTrace();
}
}
System.exit(r.getFailureCount());
} catch( Throwable t ) {
t.printStackTrace();
System.exit(1);
}
}
public static void userMain(String[] args) {
H2O.main(args);
TestUtil.stall_till_cloudsize(NODES);
List<Class> tests = new ArrayList<Class>();
// Classes to test:
// tests = JUnitRunner.all();
// Neural Net - deprecated
// tests.add(NeuralNetSpiralsTest.class); //compare NeuralNet vs reference
// tests.add(NeuralNetIrisTest.class); //compare NeuralNet vs reference
// Chunk tests
// tests.add(C0LChunkTest.class);
// tests.add(C0DChunkTest.class);
// tests.add(C1ChunkTest.class);
// tests.add(C1NChunkTest.class);
// tests.add(C1SChunkTest.class);
// tests.add(C2ChunkTest.class);
// tests.add(C2SChunkTest.class);
// tests.add(C4ChunkTest.class);
// tests.add(C4FChunkTest.class);
// tests.add(C4SChunkTest.class);
// tests.add(C8ChunkTest.class);
// tests.add(C8DChunkTest.class);
// tests.add(C16ChunkTest.class);
// tests.add(CBSChunkTest.class);
// tests.add(CX0ChunkTest.class);
// tests.add(CXIChunkTest.class);
// tests.add(CXDChunkTest.class);
// tests.add(VecTest.class);
// Deep Learning tests
// tests.add(DeepLearningVsNeuralNet.class); //only passes for NODES=1, not clear why
// tests.add(DeepLearningAutoEncoderTest.class); //test Deep Learning convergence
// tests.add(DeepLearningAutoEncoderCategoricalTest.class); //test Deep Learning
// convergence
// tests.add(DeepLearningSpiralsTest.class); //test Deep Learning convergence
// tests.add(DeepLearningIrisTest.Short.class); //compare Deep Learning vs reference
// tests.add(DeepLearningIrisTest.Long.class); //compare Deep Learning vs reference
tests.add(DeepLearningProstateTest.Short.class); // test Deep Learning
// tests.add(DeepLearningMissingTest.class); //test Deep Learning
// tests.add(DeepLearningProstateTest.Long.class); //test Deep Learning
// tests.add(NeuronsTest.class); //test Deep Learning
// tests.add(MRUtilsTest.class); //test MR sampling/rebalancing
// tests.add(DropoutTest.class); //test NN Dropput
// tests.add(ParserTest2.class);
// tests.add(ParserTest2.ParseAllSmalldata.class);
// tests.add(KMeans2Test.class);
// tests.add(KMeans2RandomTest.class);
// tests.add(GLMRandomTest.Short.class);
// tests.add(SpeeDRFTest.class);
// tests.add(SpeeDRFTest2.class);
//// tests.add(GLMTest2.class);
// tests.add(DRFTest.class);
// tests.add(DRFTest2.class);
// tests.add(GBMTest.class);
// tests.add(KMeans2Test.class);
// tests.add(PCATest.class);
// tests.add(NetworkTestTest.class);
// Uncomment this to sleep here and use the browser.
// try { Thread.sleep(10000000); } catch (Exception _) {}
JUnitCore junit = new JUnitCore();
junit.addListener(new LogListener());
Result result = junit.run(tests.toArray(new Class[0]));
if (result.getFailures().size() == 0) {
Log.info("SUCCESS!");
System.exit(0);
} else {
Log.info("FAIL!");
System.exit(1);
}
}
