def test_nfoldcrossvalidation_testall():
N = 121
C = test_classifier(N)
features = np.arange(N)
labels = np.zeros(N)
cmat,clabels = nfoldcrossvalidation(features, labels, classifier=C)
assert np.all(C.tested)
def test_predictions():
np.random.seed(222)
D = np.random.rand(100,10)
D[:40] += np.random.rand(40,10)**2
labels = [0] * 40 + [1] * 60
cmat,_,predictions = nfoldcrossvalidation(D, labels, classifier=fast_classifier(), return_predictions=1)
assert np.all((predictions == 0)|(predictions == 1))
assert cmat.trace() == np.sum(predictions == labels)
def test_nfoldcrossvalidation_simple_list():
from milksets import wine
features, labels = wine.load()
features = features[::2]
labels = labels[::2]
cmat,clabels = nfoldcrossvalidation(list(features), list(labels), classifier=fast_classifier())
assert cmat.shape == (3,3)
assert len(clabels) == 3
def test_stringlabels():
np.random.seed(222)
D = np.random.rand(100,10)
D[:40] += np.random.rand(40,10)**2
labelnames = ['one'] * 40 + ['two'] * 60
cmat,Lo = nfoldcrossvalidation(D, labelnames, classifier=fast_classifier())
assert Lo[0] in labelnames
assert Lo[1] in labelnames
assert Lo[0] != Lo[1] in labelnames
