TST Quick fix for test_classification_report_dictionary_output on 32 bits
@@ -140,7 +140,14 @@ def test_classification_report_dictionary_output():
y_true, y_pred, labels=np.arange(len(iris.target_names)),
target_names=iris.target_names, output_dict=True)
- assert_dict_equal(report, expected_report)
+ # assert the 2 dicts are equal.
+ assert(report.keys() == expected_report.keys())
+ for key in expected_report:
+ assert report[key].keys() == expected_report[key].keys()
+ for metric in expected_report[key]:
+ assert_almost_equal(expected_report[key][metric],
+ report[key][metric])
+
assert type(expected_report['setosa']['precision']) == float
assert type(expected_report['macro avg']['precision']) == float
assert type(expected_report['setosa']['support']) == int
