Exemplo n.º 1
0
def test_constraints_object_creation_from_string_without_customizations():
    with open(os.path.join(tests.integ.DATA_DIR, "monitor/constraints.json"), "r") as f:
        file_body = f.read()

    constraints = Constraints.from_string(constraints_file_string=file_body)

    assert constraints.file_s3_uri.startswith("s3://")
    assert constraints.file_s3_uri.endswith("constraints.json")

    assert constraints.body_dict["monitoring_config"]["evaluate_constraints"] == "Enabled"
Exemplo n.º 2
0
def test_constraints_object_creation_from_string_with_customizations(
    sagemaker_session, monitoring_files_kms_key
):
    with open(os.path.join(tests.integ.DATA_DIR, "monitor/constraints.json"), "r") as f:
        file_body = f.read()

    constraints = Constraints.from_string(
        constraints_file_string=file_body,
        kms_key=monitoring_files_kms_key,
        file_name="constraints.json",
        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"