code: def test_median_axis2(self, xp, dtype):
      a = testing.shaped_random((3, 4, 5), xp, dtype)
      return xp.median(a, axis=2)
  
imports: import math
  import numpy
  import pytest
  import cupy
  from cupy import testing
label_desc: Use image node_id 3 for calling the TestMedian obj's underlying member method code with example usage: obj.test_median_axis2(xp, dtype) and returns: xp