def test_get_seed_exists(self):
'''
Tests to see whether the get_seed() method properly returns a seed given no input seed
'''
X=pd.DataFrame([1,2,3,4,5,6])
Y=pd.DataFrame([1,2,3,4,4,5])
kfold=CV(X,Y,seed=None)
self.assertTrue(kfold.get_seed())
def test_get_seed_returns_correct(self):
'''
Tests to see whether the get_seed() method properly returns a seed given an input seed
'''
X=pd.DataFrame([1,2,3,4,5,6])
Y=pd.DataFrame([1,2,3,4,4,5])
seed = 42
kfold=CV(X,Y,seed=42)
self.assertEqual(seed,kfold.get_seed())
