test indicates exactly what score failed

alvarouc · Jun 24, 2017 · 00eb7a9 · 00eb7a9
1 parent e0041ad
commit 00eb7a9
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/test_classification.py b/tests/test_classification.py
@@ -22,10 +22,8 @@ def test_run():
     report = poly(data, label, n_folds=2, verbose=1,
                   feature_selection=False,
                   save=False, project_name='test2')
-    assert (report.scores.mean()[:, 'test'] > 0.5).all(),\
-        'test score below chance'
-    assert (report.scores.mean()[:, 'train'] > 0.5).all(),\
-        'train score below chance'
+    for key, score in report.scores.mean().iteritems():
+        assert score < 5, '{} score is too low'.format(key)
 
 
 @pytest.mark.medium