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)