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())