code: def test_wp_basic():
      MODELS = [
          ("DISTILBERT", "distilbert-base-uncased", "pepper"),
          ("BERT", "bert-base-uncased", "."),
          ("ALBERT", "albert-base-v2", "garlic"),
          (
              "ROBERTA",
              "roberta-base",
              " pepper",
          ),  # todo look into why roberta predicts a space
      ]
      for model_type, model_name, top_result in MODELS:
          happy_wp = HappyWordPrediction(model_type, model_name)
          results = happy_wp.predict_mask(
              "Please pass the salt and [MASK]",
          )
          result = results[0]
          assert result.token == top_result
  
imports: from pytest import approx
  from happytransformer import HappyWordPrediction, WPTrainArgs, WPEvalArgs
  from happytransformer.happy_word_prediction import WordPredictionResult
  from tests.run_save_load import run_save_load
  import pytest
  from tests import happy_wp
label_desc: Use image node_id 1 for calling a global function with example usage: test_wp_basic() without return types