def test_constraints_object_creation_from_s3_uri_without_customizations(
sagemaker_session):
with open(os.path.join(tests.integ.DATA_DIR, "monitor/constraints.json"),
"r") as f:
file_body = f.read()
file_name = "constraints.json"
desired_s3_uri = os.path.join(
"s3://",
sagemaker_session.default_bucket(),
"integ-test-test-monitoring-files",
str(uuid.uuid4()),
file_name,
)
s3_uri = S3Uploader.upload_string_as_file_body(
body=file_body, desired_s3_uri=desired_s3_uri)
constraints = Constraints.from_s3_uri(constraints_file_s3_uri=s3_uri)
assert constraints.file_s3_uri.startswith("s3://")
assert constraints.file_s3_uri.endswith("constraints.json")
assert constraints.body_dict["monitoring_config"][
"evaluate_constraints"] == "Enabled"
def test_constraints_object_creation_from_file_path_with_customizations(
sagemaker_session, monitoring_files_kms_key
):
constraints = Constraints.from_file_path(
constraints_file_path=os.path.join(tests.integ.DATA_DIR, "monitor/constraints.json"),
kms_key=monitoring_files_kms_key,
sagemaker_session=sagemaker_session,
)
assert constraints.file_s3_uri.startswith("s3://")
assert constraints.file_s3_uri.endswith("constraints.json")
assert constraints.body_dict["monitoring_config"]["evaluate_constraints"] == "Enabled"
constraints.set_monitoring(False)
assert constraints.body_dict["monitoring_config"]["evaluate_constraints"] == "Disabled"
constraints.set_monitoring(True, "message")
assert (
constraints.body_dict["features"][0]["string_constraints"]["monitoring_config_overrides"][
"evaluate_constraints"
]
== "Enabled"
)
constraints.set_monitoring(True, "second_message")
assert (
constraints.body_dict["features"][0]["string_constraints"]["monitoring_config_overrides"][
"evaluate_constraints"
]
== "Enabled"
)
constraints.save()
new_constraints = Constraints.from_s3_uri(constraints.file_s3_uri)
assert new_constraints.body_dict["monitoring_config"]["evaluate_constraints"] == "Disabled"
