def METHOD_NAME(self):
    for model_class_name in self.all_model_classes:
        if model_class_name == FlaxElectraForMaskedLM:
            model = model_class_name.from_pretrained("google/electra-small-generator")
        else:
            model = model_class_name.from_pretrained("google/electra-small-discriminator")
        outputs = model(np.ones((1, 1)))
        self.assertIsNotNone(outputs)