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"