def METHOD_NAME(self):
    api = InferenceApi("bert-base-uncased")
    inputs = "Hi, I think [MASK] is cool"
    results = api(inputs)
    self.assertIsInstance(results, list)
    result = results[0]
    self.assertIsInstance(result, dict)
    self.assertTrue("sequence" in result)
    self.assertTrue("score" in result)