def test_constraint_violations_object_creation_from_s3_uri_without_customizations(
sagemaker_session):
with open(
os.path.join(tests.integ.DATA_DIR,
"monitor/constraint_violations.json"), "r") as f:
file_body = f.read()
file_name = "constraint_violations.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,
session=sagemaker_session)
constraint_violations = ConstraintViolations.from_s3_uri(
constraint_violations_file_s3_uri=s3_uri,
sagemaker_session=sagemaker_session)
assert constraint_violations.file_s3_uri.startswith("s3://")
assert constraint_violations.file_s3_uri.endswith(
"constraint_violations.json")
assert constraint_violations.body_dict["violations"][0][
"feature_name"] == "store_and_fwd_flag"
def test_constraint_violations_object_creation_from_file_path_without_customizations():
constraint_violations = ConstraintViolations.from_file_path(
constraint_violations_file_path=os.path.join(
tests.integ.DATA_DIR, "monitor/constraint_violations.json"
)
)
assert constraint_violations.file_s3_uri.startswith("s3://")
assert constraint_violations.file_s3_uri.endswith("constraint_violations.json")
assert constraint_violations.body_dict["violations"][0]["feature_name"] == "store_and_fwd_flag"
def test_constraint_violations_object_creation_from_string_without_customizations():
with open(os.path.join(tests.integ.DATA_DIR, "monitor/constraint_violations.json"), "r") as f:
file_body = f.read()
constraint_violations = ConstraintViolations.from_string(
constraint_violations_file_string=file_body
)
assert constraint_violations.file_s3_uri.startswith("s3://")
assert constraint_violations.file_s3_uri.endswith("constraint_violations.json")
assert constraint_violations.body_dict["violations"][0]["feature_name"] == "store_and_fwd_flag"
def test_constraint_violations_object_creation_from_file_path_with_customizations(
sagemaker_session, monitoring_files_kms_key):
constraint_violations = ConstraintViolations.from_file_path(
constraint_violations_file_path=os.path.join(
tests.integ.DATA_DIR, "monitor/constraint_violations.json"),
kms_key=monitoring_files_kms_key,
sagemaker_session=sagemaker_session,
)
assert constraint_violations.file_s3_uri.startswith("s3://")
assert constraint_violations.file_s3_uri.endswith(
"constraint_violations.json")
assert constraint_violations.body_dict["violations"][0][
"feature_name"] == "store_and_fwd_flag"
