def test_test_data_uri(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.test_data_uri == _TEST_TEST_DATA_URI
def test_test_data_uri_none(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.test_data_uri is None
def test_tf_config_none(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.tf_config is None
def test_validation_data_uri(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.validation_data_uri == _TEST_VALIDATION_DATA_URI
def test_cluster_spec_none(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.cluster_spec is None
def test_tf_config(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.tf_config == json.loads(_TEST_CLUSTER_SPEC)
def test_tensorboard_log_dir_none(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.tensorboard_log_dir is None
def test_cluster_spec(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.cluster_spec == json.loads(_TEST_CLUSTER_SPEC)
def test_tensorboard_log_dir(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.tensorboard_log_dir == _TEST_TENSORBOARD_LOG_DIR
def test_checkpoint_dir_none(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.checkpoint_dir is None
def test_checkpoint_dir(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.checkpoint_dir == _TEST_CHECKPOINT_DIR
def test_model_dir_none(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.model_dir is None
def test_model_dir(self):
env_vars = training_utils.EnvironmentVariables()
assert env_vars.model_dir == _TEST_MODEL_DIR