code: def test_wp_save():
      nft_happy_wp.save("model/")
      result_before = nft_happy_wp.predict_mask(
          "I think therefore I [MASK]"
      )
  
      happy = HappyWordPrediction(model_name="model/")
      result_after = happy.predict_mask("I think therefore I [MASK]")
  
      assert result_before[0].token == result_after[0].token
  
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 12 for calling a global function with example usage: test_wp_save() without return types