def test_rdd_double_h2o_frame(self): hc = self._hc rdd = self._sc.parallelize([0.5, 1.3333333333, 178]) h2o_frame = hc.as_h2o_frame(rdd) self.assertEquals(h2o_frame[0, 0], 0.5, "Value should match") self.assertEquals(h2o_frame[1, 0], 1.3333333333, "Value should match") test_utils.asert_h2o_frame(self, h2o_frame, rdd)
def test_rdd_bool_to_h2o_frame(self): hc = self._hc rdd = self._sc.parallelize([True, False, True, True, False]) h2o_frame = hc.as_h2o_frame(rdd) self.assertEquals(h2o_frame[0, 0], 1, "Value should match") self.assertEquals(h2o_frame[1, 0], 0, "Value should match") test_utils.asert_h2o_frame(self, h2o_frame, rdd)
def test_rdd_str_h2o_frame(self): hc = self._hc rdd = self._sc.parallelize(["a", "b", "c"]) h2o_frame = hc.as_h2o_frame(rdd) self.assertEquals(h2o_frame[0, 0], "a", "Value should match") self.assertEquals(h2o_frame[2, 0], "c", "Value should match") test_utils.asert_h2o_frame(self, h2o_frame, rdd)
def test_rdd_int_h2o_frame(self): hc = self._hc rdd = self._spark.sparkContext.parallelize( [num for num in range(0, 100)]) h2o_frame = hc.as_h2o_frame(rdd) self.assertEquals(h2o_frame[0, 0], 0, "Value should match") test_utils.asert_h2o_frame(self, h2o_frame, rdd)
def test_rdd_double_h2o_frame(self): hc = self._hc rdd = self._spark.sparkContext.parallelize([0.5,1.3333333333,178]) h2o_frame = hc.as_h2o_frame(rdd) self.assertEquals(h2o_frame[0,0],0.5,"Value should match") self.assertEquals(h2o_frame[1,0],1.3333333333,"Value should match") test_utils.asert_h2o_frame(self, h2o_frame, rdd)
def test_rdd_str_h2o_frame(self): hc = self._hc rdd = self._spark.sparkContext.parallelize(["a","b","c"]) h2o_frame = hc.as_h2o_frame(rdd) self.assertEquals(h2o_frame[0,0],"a","Value should match") self.assertEquals(h2o_frame[2,0],"c","Value should match") test_utils.asert_h2o_frame(self,h2o_frame,rdd)
def test_rdd_bool_to_h2o_frame(self): hc = self._hc rdd = self._spark.sparkContext.parallelize([True, False, True, True, False]) h2o_frame = hc.as_h2o_frame(rdd) self.assertEquals(h2o_frame[0,0],1,"Value should match") self.assertEquals(h2o_frame[1,0],0,"Value should match") test_utils.asert_h2o_frame(self,h2o_frame,rdd)
def test_rdd_int_h2o_frame(self): hc = self._hc rdd = self._spark.sparkContext.parallelize([num for num in range(0,100)]) h2o_frame = hc.as_h2o_frame(rdd) self.assertEquals(h2o_frame[0,0], 0, "Value should match") test_utils.asert_h2o_frame(self,h2o_frame,rdd)