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"