From f18a5f4511c3d133feb91fb47ad5797ecec23697 Mon Sep 17 00:00:00 2001 From: David Marek Date: Sun, 15 Apr 2012 12:45:15 +0200 Subject: [PATCH] TEST: assert_equal instead of assert --- sklearn/tests/test_cross_validation.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/sklearn/tests/test_cross_validation.py b/sklearn/tests/test_cross_validation.py index 98cc3a66716a4..9b33f5316dbe5 100644 --- a/sklearn/tests/test_cross_validation.py +++ b/sklearn/tests/test_cross_validation.py @@ -4,7 +4,7 @@ import numpy as np from scipy.sparse import coo_matrix -from nose.tools import assert_true +from nose.tools import assert_true, assert_equal from nose.tools import assert_raises from ..base import BaseEstimator @@ -177,11 +177,12 @@ def test_shuffle_split_warnings(): cval.ShuffleSplit(10, 3, train_fraction=0.1) cval.train_test_split(range(3), test_fraction=0.1) cval.train_test_split(range(3), train_fraction=0.1) - assert len(warn_queue) == 4 - assert warn_queue[0] == expected_message[0] - assert warn_queue[1] == expected_message[1] - assert warn_queue[2] == expected_message[0] - assert warn_queue[3] == expected_message[1] + + assert_equal(len(warn_queue), 4) + assert_equal(warn_queue[0], expected_message[0]) + assert_equal(warn_queue[1], expected_message[1]) + assert_equal(warn_queue[2], expected_message[0]) + assert_equal(warn_queue[3], expected_message[1]) # restore default behavior warnings.warn = warnings_warn