use of water.rapids.Val in project h2o-3 by h2oai.
the class AstSumAxisTest method testColumnwisesumWithoutNaRm.
@Test
public void testColumnwisesumWithoutNaRm() {
Frame fr = register(new Frame(Key.<Frame>make(), ar("I", "D", "DD", "DN", "T", "S", "C"), aro(vi1, vd1, vd2, vd3, vt1, vs1, vc2)));
Val val1 = Rapids.exec("(sumaxis " + fr._key + " 0 0)");
assertTrue(val1 instanceof ValFrame);
Frame res = register(val1.getFrame());
assertArrayEquals(fr.names(), res.names());
assertArrayEquals(ar(Vec.T_NUM, Vec.T_NUM, Vec.T_NUM, Vec.T_NUM, Vec.T_TIME, Vec.T_NUM, Vec.T_NUM), res.types());
assertRowFrameEquals(ard(0.0, 20.0, 3.0, Double.NaN, 50000150.0, Double.NaN, Double.NaN), res);
}
use of water.rapids.Val in project h2o-3 by h2oai.
the class AstSumAxisTest method testColumnwisesumBinaryVec.
@Test
public void testColumnwisesumBinaryVec() {
assertTrue(vc1.isBinary() && !vc2.isBinary());
Frame fr = register(new Frame(Key.<Frame>make(), ar("C1", "C2"), aro(vc1, vc2)));
Val val = Rapids.exec("(sumaxis " + 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(3.0, Double.NaN), res);
}
use of water.rapids.Val in project h2o-3 by h2oai.
the class AstSumAxisTest method testRowwisesumOnFrameWithTimeandNumericColumn.
@Test
public void testRowwisesumOnFrameWithTimeandNumericColumn() {
Frame fr = register(new Frame(Key.<Frame>make(), ar("t1", "i1"), aro(vt1, vi1)));
Val val = Rapids.exec("(sumaxis " + fr._key + " 1 1)");
assertTrue(val instanceof ValFrame);
Frame res = register(val.getFrame());
assertColFrameEquals(ard(-1, -2, 0, 2, 1), res);
}
use of water.rapids.Val in project h2o-3 by h2oai.
the class AstSumAxisTest method testRowwisesumOnFrameWithNonnumericColumnsOnly.
@Test
public void testRowwisesumOnFrameWithNonnumericColumnsOnly() {
Frame fr = register(new Frame(Key.<Frame>make(), ar("c1", "s1"), aro(vc2, vs1)));
Val val = Rapids.exec("(sumaxis " + fr._key + " 1 1)");
assertTrue(val instanceof ValFrame);
Frame res = register(val.getFrame());
assertEquals("Unexpected column name", "sum", 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);
}
use of water.rapids.Val in project h2o-3 by h2oai.
the class AstSumAxisTest method testRowwisesumWithoutNaRm.
@Test
public void testRowwisesumWithoutNaRm() {
Frame fr = register(new Frame(Key.<Frame>make(), ar("i1", "d1", "d2", "d3"), aro(vi1, vd1, vd2, vd3)));
Val val = Rapids.exec("(sumaxis " + fr._key + " 0 1)");
assertTrue(val instanceof ValFrame);
Frame res = register(val.getFrame());
assertColFrameEquals(ard(1.7, 2.9, Double.NaN, 10.3, Double.NaN), res);
assertEquals("sum", res.name(0));
}
Aggregations