def test_CV_split(self):
'''
Tests the split method to see whether the splits are of the right shape and length
'''
X=pd.DataFrame([1,2,3,4,5,6])
Y=pd.DataFrame([1,2,3,4,4,5])
n_splits = 3
# Initialize CV
kfold=CV(X,Y)
# Do splits of 3
self.assertEqual(len(kfold.split(n_splits=n_splits)[0]), 3)
self.assertEqual(len(kfold.split(n_splits=n_splits)[1]), 3)
