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));
}
use of water.rapids.Val in project h2o-3 by h2oai.
the class AstMatchTest method testMatchCatList.
@Test
public void testMatchCatList() throws Exception {
final Frame data = makeTestFrame();
Frame output = null;
try {
String rapids = "(tmp= tst (match (cols data [2]) [\"cD\",\"cC\",\"cB\"] -1 ignored))";
Val val = Rapids.exec(rapids);
output = val.getFrame();
assertVecEquals(data.vec(0), output.vec(0), 0.0);
} finally {
data.delete();
if (output != null) {
output.delete();
}
}
}
use of water.rapids.Val in project h2o-3 by h2oai.
the class AstRectangleConditionalAssignTest method testConditionalAssignUUID.
@Test
public void testConditionalAssignUUID() {
UUID newUUID = UUID.randomUUID();
Frame fr = makeTestFrame();
Vec expected = uvec(new UUID(10, 1), newUUID, new UUID(30, 3), newUUID, new UUID(50, 5));
try {
Val val = Rapids.exec("(tmp= py_1 (:= data \"" + newUUID.toString() + "\" 2 (== (cols_py data 4) \"a\")))");
if (val instanceof ValFrame) {
Frame fr2 = val.getFrame();
assertUUIDVecEquals(expected, fr2.vec(2));
fr2.remove();
}
} finally {
fr.remove();
expected.remove();
}
}
use of water.rapids.Val in project h2o-3 by h2oai.
the class AstRectangleConditionalAssignTest method testConditionalAssignNumber.
@Test
public void testConditionalAssignNumber() {
Frame fr = makeTestFrame();
Vec expected = dvec(11.2, -1, 33.6, -1, 56.0);
try {
Val val = Rapids.exec("(tmp= py_1 (:= data -1 1 (== (cols_py data 4) \"a\")))");
if (val instanceof ValFrame) {
Frame fr2 = val.getFrame();
assertVecEquals(expected, fr2.vec(1), 0.00001);
fr2.remove();
}
} finally {
fr.remove();
expected.remove();
}
}
use of water.rapids.Val in project h2o-3 by h2oai.
the class AstRectangleConditionalAssignTest method testConditionalAssignDense.
@Test
public void testConditionalAssignDense() {
Frame fr = makeTestFrame();
Vec expected = ivec(-1, -1, -1, -1, -1);
try {
Val val = Rapids.exec("(tmp= py_1 (:= data -1 2 (> (cols_py data 0) 0)))");
if (val instanceof ValFrame) {
Frame fr2 = val.getFrame();
assertVecEquals(expected, fr2.vec(2), 0.00001);
fr2.remove();
}
} finally {
fr.remove();
expected.remove();
}
}
Aggregations