def METHOD_NAME(self):
    for model_class_name in self.all_model_classes:
        model = model_class_name.from_pretrained("roberta-base", from_pt=True)
        outputs = model(np.ones((1, 1)))
        self.assertIsNotNone(outputs)