Skip to content

Commit fe50e05

Browse files
committed
updated unit tests to align with code changes
1 parent 8b72a60 commit fe50e05

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unit/test_write_json_files.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -262,15 +262,15 @@ def test_write_model_properties_json():
262262
target_values=None,
263263
)
264264
assert "ModelProperties.json" in prop_dict
265-
assert json.loads(prop_dict["ModelProperties.json"])["function"] == "Prediction"
265+
assert json.loads(prop_dict["ModelProperties.json"])["function"] == "prediction"
266266
assert json.loads(prop_dict["ModelProperties.json"])["targetLevel"] == "Interval"
267267
assert json.loads(prop_dict["ModelProperties.json"])["targetEvent"] == ""
268268
assert json.loads(prop_dict["ModelProperties.json"])["eventProbVar"] == ""
269269

270270
prop_dict = jf.write_model_properties_json(
271271
model_name="Test_Model", target_variable="BAD", target_values=[1, 0]
272272
)
273-
assert json.loads(prop_dict["ModelProperties.json"])["function"] == "Classification"
273+
assert json.loads(prop_dict["ModelProperties.json"])["function"] == "classification"
274274
assert json.loads(prop_dict["ModelProperties.json"])["targetLevel"] == "Binary"
275275
assert json.loads(prop_dict["ModelProperties.json"])["targetEvent"] == "1"
276276
assert json.loads(prop_dict["ModelProperties.json"])["eventProbVar"] == "P_1"

0 commit comments

Comments
 (0)