code: def test_stamp_nan_inf_A_B_join(
      T_A, T_B, substitute_A, substitute_B, substitution_locations
  ):
      m = 3
  
      T_A_sub = T_A.copy()
      T_B_sub = T_B.copy()
  
      for substitution_location_B in substitution_locations:
          for substitution_location_A in substitution_locations:
              T_A_sub[:] = T_A[:]
              T_B_sub[:] = T_B[:]
              T_A_sub[substitution_location_A] = substitute_A
              T_B_sub[substitution_location_B] = substitute_B
  
              ref_mp = naive.stump(
                  T_A_sub, m, T_B=T_B_sub, row_wise=True
              )
              comp_mp = stamp.stamp(T_A_sub, T_B_sub, m)
              naive.replace_inf(ref_mp)
              naive.replace_inf(comp_mp)
              npt.assert_almost_equal(ref_mp[:, :2], comp_mp)
  
file_path: stumpy/tests/test_stamp.py
imports: import functools
  import naive
  import numpy
  import numpy.testing
  import pytest
  from stumpy import core, stamp
label_desc: Use image node_id 6 for calling a global function with example usage: test_stamp_nan_inf_A_B_join(T_A, T_B, substitute_A, substitute_B, substitution_locations) without return types