code: def test_extraction_words_ids():
      """
      Assert that input words are split.
      Assert that indexes are updated and extractable.
      """
  
      lda = preprocessing.LDA(2, number_of_documents=5, seed=42)
  
      word_indexes_list = []
  
      for doc in DOC_SET:
          words = doc.split(" ")
  
          lda._update_indexes(word_list=words)
  
          word_indexes_list.append(
              [lda.word_to_index[word] for word in words]
          )
  
      assert word_indexes_list == [
          [1, 2],
          [1, 3, 4],
          [1, 2, 5],
          [1, 3],
          [1, 2, 6],
      ]
  
file_path: river/river/preprocessing/test_lda.py
imports: from __future__ import annotations
  import numpy
  from river import preprocessing
label_desc: Use image node_id 1 for calling a global function with example usage: test_extraction_words_ids() without return types