def setUp(self): self.env = StreamExecutionEnvironment.get_execution_environment() self._load_dependency_jars() config = Configuration( j_configuration=get_j_env_configuration(self.env._j_stream_execution_environment)) config.set_boolean("execution.checkpointing.checkpoints-after-tasks-finish.enabled", True) self.env.set_parallelism(4) self.env.enable_checkpointing(100) self.env.set_restart_strategy(RestartStrategies.no_restart()) self.t_env = StreamTableEnvironment.create(self.env) self.temp_dir = tempfile.mkdtemp()
def test_to_dict(self): conf = Configuration() conf.set_string("k1", "v1") conf.set_integer("k2", 1) conf.set_float("k3", 1.2) conf.set_boolean("k4", True) target_dict = conf.to_dict() self.assertEqual(target_dict, { "k1": "v1", "k2": "1", "k3": "1.2", "k4": "true" })
def test_add_all_to_dict(self): conf = Configuration() conf.set_string("k1", "v1") conf.set_integer("k2", 1) conf.set_float("k3", 1.2) conf.set_boolean("k4", True) conf.set_bytearray("k5", bytearray([1, 2, 3])) target_dict = dict() conf.add_all_to_dict(target_dict) self.assertEqual(target_dict, { "k1": "v1", "k2": 1, "k3": 1.2, "k4": True, "k5": bytearray([1, 2, 3]) })
def test_getters_and_setters(self): conf = Configuration() conf.set_string("str", "v1") conf.set_integer("int", 2) conf.set_boolean("bool", True) conf.set_float("float", 0.5) conf.set_bytearray("bytearray", bytearray([1, 2, 3])) str_value = conf.get_string("str", "") int_value = conf.get_integer("int", 0) bool_value = conf.get_boolean("bool", False) float_value = conf.get_float("float", 0) bytearray_value = conf.get_bytearray("bytearray", bytearray()) self.assertEqual(str_value, "v1") self.assertEqual(int_value, 2) self.assertEqual(bool_value, True) self.assertEqual(float_value, 0.5) self.assertEqual(bytearray_value, bytearray([1, 2, 3]))