code: def test_stump_A_B_join_KNN(T_A, T_B):
      m = 3
      for k in range(2, 4):
          ref_mp = naive.stump(T_A, m, T_B=T_B, k=k)
          comp_mp = stump(T_A, m, T_B, ignore_trivial=False, k=k)
          naive.replace_inf(ref_mp)
          naive.replace_inf(comp_mp)
          npt.assert_almost_equal(ref_mp, comp_mp)
  
          comp_mp = stump(
              pd.Series(T_A),
              m,
              pd.Series(T_B),
              ignore_trivial=False,
              k=k,
          )
          naive.replace_inf(comp_mp)
          npt.assert_almost_equal(ref_mp, comp_mp)
  
imports: import functools
  import naive
  import numpy
  import numpy.testing
  import pandas
  import pytest
  from stumpy import config, stump
label_desc: Use image node_id 13 for calling a global function with example usage: test_stump_A_B_join_KNN(T_A, T_B) without return types