class TolerantEvaluatorTransformTests(TestCase):
    def test_errors_are_raised(self):
        self.ev = TolerantFeatureEvaluator([BrokenFeature, DumbFeatureA])
        self.ev.fit([])

        def transform():
            list(self.ev.transform(SAMPLES))  # force generation

        self.assertRaises(RuntimeError, transform)
class TolerantEvaluatorTransformTests(TestCase):

    def test_errors_are_raised(self):
        self.ev = TolerantFeatureEvaluator([BrokenFeature, DumbFeatureA])
        self.ev.fit([])

        def transform():
            list(self.ev.transform(SAMPLES))  # force generation
        self.assertRaises(RuntimeError, transform)
class TolerantEvaluatorFitTests(TestCase, TolerantFittingCases):
    fit_method_name = 'fit'

    def test_returns_itself(self):
        self.ev = TolerantFeatureEvaluator([DumbFeatureA])
        self.assertEqual(self.ev.fit([]), self.ev)
class TolerantEvaluatorFitTests(TestCase, TolerantFittingCases):
    fit_method_name = 'fit'

    def test_returns_itself(self):
        self.ev = TolerantFeatureEvaluator([DumbFeatureA])
        self.assertEqual(self.ev.fit([]), self.ev)