def testDefaultFitMultiple(self):
N = 4
data = MockDataset()
estimator = MockEstimator()
params = [{estimator.fake: i} for i in range(N)]
modelIter = estimator.fitMultiple(data, params)
indexList = []
for index, model in modelIter:
self.assertEqual(model.getFake(), index)
indexList.append(index)
self.assertEqual(sorted(indexList), list(range(N)))
def testDefaultFitMultiple(self):
N = 4
data = MockDataset()
estimator = MockEstimator()
params = [{estimator.fake: i} for i in range(N)]
modelIter = estimator.fitMultiple(data, params)
indexList = []
for index, model in modelIter:
self.assertEqual(model.getFake(), index)
indexList.append(index)
self.assertEqual(sorted(indexList), list(range(N)))
class EstimatorTest(unittest.TestCase):
def setUp(self):
self.estimator = MockEstimator()
self.data = MockDataset()
def test_fit_invalid_params(self):
invalid_type_parms = ""
self.assertRaises(TypeError, self.estimator.fit, self.data, invalid_type_parms)
def testDefaultFitMultiple(self):
N = 4
params = [{self.estimator.fake: i} for i in range(N)]
modelIter = self.estimator.fitMultiple(self.data, params)
indexList = []
for index, model in modelIter:
self.assertEqual(model.getFake(), index)
indexList.append(index)
self.assertEqual(sorted(indexList), list(range(N)))