package hex.drf;

import org.junit.BeforeClass;
import org.junit.Test;
import water.TestUtil;
import water.fvec.Frame;
import water.UKV;

public class DRFTest2 extends TestUtil {

  @BeforeClass public static void stall() { stall_till_cloudsize(1); }

  static final String[] s(String...arr)  { return arr; }
  static final long[]   a(long ...arr)   { return arr; }
  static final long[][] a(long[] ...arr) { return arr; }

  // A bigger DRF test, useful for tracking memory issues.
  @Test public void testAirlines() throws Throwable {
    new DRFTest().basicDRF(
        //
        //"../demo/c5/row10000.csv.gz", "c5.hex", null, null, 
        "../datasets/UCI/UCI-large/covtype/covtype.data", "covtype.hex", null, null, 
        //"./smalldata/iris/iris_wheader.csv", "iris.hex", null, null, 
        new DRFTest.PrepData() { @Override int prep(Frame fr) { return fr.numCols()-1; } },
        10/*ntree*/,
        a( a(154871,  52304,    18,    0,   24,   15,   935),  
           a( 34958, 240871,  1423,   21,  243,  656,    56),  
           a(     4,   3836, 30040,  203,    1, 1049,     0),  
           a(     0,      8,   640, 1960,    0,   81,     0),  
           a(   229,   6824,   126,    0, 2137,   17,     0),  
           a(    21,   4278,  5353,   91,    2, 7332,     0),  
           a(  7170,    297,     0,    0,    0,    0, 12718)),
        s("1", "2", "3", "4", "5", "6", "7"),
        50/*max_depth*/);
  }
}

@@ -921,27 +921,27 @@
  a(1
-54871,  52304
+99145,   7055
 ,    
-1
+ 
 8,  
@@ -950,28 +950,28 @@
 0,  
- 24
+115
 ,   
-15
+ 22
 ,   
-935
+489
 ), 
- 
 %0A   
@@ -985,57 +985,57 @@
  a( 
-34958, 240871,  1423
+ 9892, 265406,   434
 ,   
-21,  243,  656
+ 3,  456,   251
 ,   
- 56
+161
 ), 
- 
 %0A   
@@ -1052,54 +1052,54 @@
     
- 4
+11
 ,   
-3836, 30040,  203,    1, 1049
+ 748, 33254,  135,   23,   636
 ,     
-0
+1
 ), 
- 
 %0A   
@@ -1125,22 +1125,22 @@
     
-8
+3
 ,   
-640, 1960
+372, 2214
 ,   
@@ -1149,10 +1149,11 @@
 ,   
-81
+ 59
 ,   
@@ -1150,33 +1150,32 @@
     59,     0), 
- 
 %0A           a(  
@@ -1179,55 +1179,55 @@
 (   
-22
+16
 9,   
-6824
+1739
 ,   1
-26
+18
 ,    
-0, 2137
+1, 7115
 ,   
-17
+ 38
 ,     
-0
+4
 ), 
- 
 %0A   
@@ -1244,42 +1244,43 @@
     
-21
+30
 ,   
-4278,  5353,   91,    2, 733
+ 654,  1458,  107,   37, 1451
 2,  
@@ -1286,17 +1286,16 @@
     0), 
- 
 %0A       
@@ -1306,20 +1306,20 @@
 a(  
-7170
+1057
 ,    
-297
+156
 ,   
@@ -1336,18 +1336,19 @@
     
-0,
+8, 
     0, 1
 2718
@@ -1347,11 +1347,11 @@
 0, 1
-271
+883
 8)),

Fix covtype depth 50 golden answers