'Based on the input data, it should dispatch the fitting process to
the appropriate learner'

def test_dispatches_to_correct_learner(self):

    DummyClassificationLearner.fit = Mock()
    DummyRegressionLearner.fit = Mock()
    fitter = DummyFitter()
    fitter(self.heart_disease)
    self.assertEqual(DummyClassificationLearner.fit.call_count, 1, 'Classification    learner    was    never    called    for    classificationproblem')
    self.assertEqual(DummyRegressionLearner.fit.call_count, 0, 'Regression    learner    was    called    for    classification    problem')
    DummyClassificationLearner.fit.reset_mock()
    DummyRegressionLearner.fit.reset_mock()
    fitter(self.housing)
    self.assertEqual(DummyRegressionLearner.fit.call_count, 1, 'Regression    learner    was    never    called    for    regression    problem')
    self.assertEqual(DummyClassificationLearner.fit.call_count, 0, 'Classification    learner    was    called    for    regression    problem')
