Search in sources :

Example 21 with Val

use of water.rapids.Val in project h2o-3 by h2oai.

the class AstMeanTest method testRowwiseMeanOnFrameWithNonnumericColumnsOnly.

@Test
public void testRowwiseMeanOnFrameWithNonnumericColumnsOnly() {
    Frame fr = register(new Frame(Key.<Frame>make(), ar("c1", "s1"), aro(vc2, vs1)));
    Val val = Rapids.exec("(mean " + fr._key + " 1 1)");
    assertTrue(val instanceof ValFrame);
    Frame res = register(val.getFrame());
    assertEquals("Unexpected column name", "mean", res.name(0));
    assertEquals("Unexpected column type", Vec.T_NUM, res.types()[0]);
    assertColFrameEquals(ard(Double.NaN, Double.NaN, Double.NaN, Double.NaN, Double.NaN), res);
}
Also used : Val(water.rapids.Val) ValFrame(water.rapids.vals.ValFrame) ValFrame(water.rapids.vals.ValFrame) Frame(water.fvec.Frame) Test(org.junit.Test)

Example 22 with Val

use of water.rapids.Val in project h2o-3 by h2oai.

the class AstMeanTest method testRowwiseMeanWithoutNaRm.

@Test
public void testRowwiseMeanWithoutNaRm() {
    Frame fr = register(new Frame(Key.<Frame>make(), ar("i1", "d1", "d2", "d3"), aro(vi1, vd1, vd2, vd3)));
    Val val = Rapids.exec("(mean " + fr._key + " 0 1)");
    assertTrue(val instanceof ValFrame);
    Frame res = register(val.getFrame());
    assertColFrameEquals(ard(1.7 / 4, 2.9 / 4, Double.NaN, 10.3 / 4, Double.NaN), res);
    assertEquals("mean", res.name(0));
}
Also used : Val(water.rapids.Val) ValFrame(water.rapids.vals.ValFrame) ValFrame(water.rapids.vals.ValFrame) Frame(water.fvec.Frame) Test(org.junit.Test)

Example 23 with Val

use of water.rapids.Val in project h2o-3 by h2oai.

the class AstMeanTest method testColumnwiseMeanBinaryVec.

@Test
public void testColumnwiseMeanBinaryVec() {
    assertTrue(vc1.isBinary() && !vc2.isBinary());
    Frame fr = register(new Frame(Key.<Frame>make(), ar("C1", "C2"), aro(vc1, vc2)));
    Val val = Rapids.exec("(mean " + fr._key + " 1 0)");
    assertTrue(val instanceof ValFrame);
    Frame res = register(val.getFrame());
    assertArrayEquals(fr.names(), res.names());
    assertArrayEquals(ar(Vec.T_NUM, Vec.T_NUM), res.types());
    assertRowFrameEquals(ard(0.6, Double.NaN), res);
}
Also used : Val(water.rapids.Val) ValFrame(water.rapids.vals.ValFrame) ValFrame(water.rapids.vals.ValFrame) Frame(water.fvec.Frame) Test(org.junit.Test)

Example 24 with Val

use of water.rapids.Val in project h2o-3 by h2oai.

the class AstMeanTest method testRowwiseMeanOnEmptyFrame.

@Test
public void testRowwiseMeanOnEmptyFrame() {
    Frame fr = register(new Frame(Key.<Frame>make()));
    Val val = Rapids.exec("(mean " + fr._key + " 0 1)");
    assertTrue(val instanceof ValFrame);
    Frame res = register(val.getFrame());
    assertEquals(res.numCols(), 0);
    assertEquals(res.numRows(), 0);
}
Also used : Val(water.rapids.Val) ValFrame(water.rapids.vals.ValFrame) ValFrame(water.rapids.vals.ValFrame) Frame(water.fvec.Frame) Test(org.junit.Test)

Example 25 with Val

use of water.rapids.Val in project h2o-3 by h2oai.

the class AstNaCntTest method testAstNaCnt.

//--------------------------------------------------------------------------------------------------------------------
// Tests
//--------------------------------------------------------------------------------------------------------------------
@Test
public void testAstNaCnt() {
    Frame f = null;
    try {
        f = ArrayUtils.frame(ar("A", "B"), ard(1.0, Double.NaN), ard(2.0, 23.3), ard(3.0, 3.3), ard(Double.NaN, 3.3));
        String x = String.format("(naCnt %s)", f._key);
        // make the call to count number of NAs in frame
        Val res = Rapids.exec(x);
        // get frame without any NAs
        //      assertEquals(f.numRows()-fNew.numRows() ,2);  // 2 rows of NAs removed.
        double[] naCnts = res.getNums();
        double totalNacnts = 0;
        for (int index = 0; index < f.numCols(); index++) {
            totalNacnts += naCnts[index];
        }
        assertEquals((int) totalNacnts, 2);
    } finally {
        if (f != null)
            f.delete();
    }
}
Also used : Val(water.rapids.Val) Frame(water.fvec.Frame) Test(org.junit.Test)

Aggregations

Val (water.rapids.Val)76 Frame (water.fvec.Frame)65 Test (org.junit.Test)56 ValFrame (water.rapids.vals.ValFrame)52 Vec (water.fvec.Vec)14 ValRow (water.rapids.vals.ValRow)8 MRTask (water.MRTask)5 Session (water.rapids.Session)5 Chunk (water.fvec.Chunk)4 TestFrameBuilder (water.fvec.TestFrameBuilder)4 ValNum (water.rapids.vals.ValNum)4 NewChunk (water.fvec.NewChunk)2 AstParameter (water.rapids.ast.AstParameter)2 AstNumList (water.rapids.ast.params.AstNumList)2 CreateFrame (hex.CreateFrame)1 GLRMParameters (hex.glrm.GLRMModel.GLRMParameters)1 FileInputStream (java.io.FileInputStream)1 InputStream (java.io.InputStream)1 ArrayList (java.util.ArrayList)1 HashMap (java.util.HashMap)1