def setUpClass(cls):
     cls._spark = SparkSession.builder.config(
         conf=unit_test_utils.get_default_spark_conf()).getOrCreate()
     unit_test_utils.set_up_class(cls)
     cls._hc = H2OContext.getOrCreate(
         cls._spark,
         H2OConf(cls._spark).set_num_of_external_h2o_nodes(2))
Пример #2
0
 def setUpClass(cls):
     cls._cloud_name = generic_test_utils.unique_cloud_name("h2o_conf_test")
     cls._spark = SparkSession.builder.config(
         conf=unit_test_utils.get_default_spark_conf().set(
             "spark.ext.h2o.cloud.name", cls._cloud_name)).getOrCreate()
     unit_test_utils.set_up_class(cls)
     h2o_conf = H2OConf(cls._spark).set_num_of_external_h2o_nodes(2)
     cls._hc = H2OContext.getOrCreate(cls._spark, h2o_conf)
Пример #3
0
 def setUpClass(cls):
     cls._spark = SparkSession.builder.config(
         conf=unit_test_utils.get_default_spark_conf().setMaster(
             "yarn-client")).getOrCreate()
     unit_test_utils.set_up_class(cls)
     cls._hc = H2OContext.getOrCreate(
         cls._spark,
         H2OConf(cls._spark).set_cluster_size(1))
Пример #4
0
 def setUpClass(cls):
     cls._spark = SparkSession.builder.config(
         conf=unit_test_utils.get_default_spark_conf().setMaster("yarn-client")).getOrCreate()
     unit_test_utils.set_up_class(cls)
     cls._hc = H2OContext.getOrCreate(cls._spark, H2OConf(cls._spark).set_num_of_external_h2o_nodes(1))