code: def test_wp_len_max_len_pad():
      train_data = "../data/wp/train-eval.csv"
      save_path = "data/wp/len-test/"
      max_length = 32
  
      args_max_len_truncation = WPTrainArgs(
          max_length=max_length, num_train_epochs=1, save_path=save_path
      )
  
      happy_wp.train(train_data, args=args_max_len_truncation)
  
      tok_data = load_from_disk(save_path)
      print(tok_data)
  
      for case in tok_data["train"]["input_ids"]:
          print(case)
          assert len(case) == max_length
  
      for case in tok_data["eval"]["input_ids"]:
          print(case)
          assert len(case) == max_length
  
file_path: happy-transformer/tests/test_tok_length.py
imports: import csv
  from happytransformer import HappyGeneration, HappyTextToText, HappyWordPrediction, GENTrainArgs, WPTrainArgs, TTTrainArgs
  from datasets import load_from_disk
  from tests import happy_gen, happy_wp, happy_tt
label_desc: Use image node_id 4 for calling a global function with example usage: test_wp_len_max_len_pad() without return types