Fix TravisCI Python2 test error (no $DISPLAY)

sergioburdisso · May 14, 2020 · 6c7d8d9 · 6c7d8d9
1 parent 0a897dd
commit 6c7d8d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_util.py b/tests/test_util.py
@@ -122,8 +122,8 @@ def test_evaluation(mocker):
 
     # test
     #   OK
-    assert Evaluation.test(clf_ml, x_data_ml, y_data_ml) == 1 / 3.
-    assert Evaluation.test(clf_ml, x_data_ml, y_data_ml, metric='exact-match') == .5
+    assert Evaluation.test(clf_ml, x_data_ml, y_data_ml, plot=PY3) == 1 / 3.
+    assert Evaluation.test(clf_ml, x_data_ml, y_data_ml, metric='exact-match', plot=PY3) == .5
     assert Evaluation.test(clf, x_data, y_data, def_cat='unknown', plot=PY3) == 1
     assert Evaluation.test(clf, x_data, y_data, def_cat='neg', plot=PY3) == 1
     assert Evaluation.test(clf, x_data, y_data, metric="f1-score", plot=PY3) == 1