code: def test_wp_top_k():
      result = nft_happy_wp.predict_mask(
          "Please pass the salt and [MASK]", top_k=2
      )
      print(result)
      answer = [
          WordPredictionResult(
              token="pepper", score=approx(0.2664579749107361, 0.01)
          ),
          WordPredictionResult(
              token="vinegar", score=approx(0.08760260790586472, 0.01)
          ),
      ]
  
      assert result == answer
  
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 3 for calling a global function with example usage: test_wp_top_k() without return types