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)