def testKFoldsRandomize(self):
kfolds = data_split.KFolds(2)
partition1 = kfolds.split(range(10))
partition2 = kfolds.split(range(10), randomize=True)
self.assertNotEqual(partition1, partition2)
self.assertItemsEqual(partition1[0][0] + partition1[0][1],
partition2[0][0] + partition2[0][1])
def testKFolds2(self):
kfolds = data_split.KFolds(2)
with self.assertRaises(ValueError):
kfolds.split(xrange(1))
self.assertSequenceEqual(kfolds.split(xrange(2)), [([1], [0]),
([0], [1])])
self.assertSequenceEqual(kfolds.split(xrange(3)), [([1, 2], [0]),
([0, 2], [1])])
def testKFoldsBadKValue(self):
with self.assertRaises(ValueError):
data_split.KFolds(1)
def testKFoldsBadKType(self):
with self.assertRaises(TypeError):
data_split.KFolds(3.0)