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); } }