def test_is_trained(self): o = NullObject() o.is_trained = True self.assertIsNone(model.requires_training_wheels(lambda x: None)(o))
def test_not_trained(self): o = NullObject() o.is_trained = False with self.assertRaises(TypeError): model.requires_training_wheels(lambda x: None)(o)