def METHOD_NAME(self):
    # Only check this for base model, not necessary for all model classes.
    # This will also help speed-up tests.
    model = FlaxBertModel.from_pretrained("bert-base-cased")
    outputs = model(np.ones((1, 1)))
    self.assertIsNotNone(outputs)