code: def test_diag_bounds(self):
      A = [[1, 2], [3, 4], [5, 6]]
      assert_equal(diag(A, k=2), [])
      assert_equal(diag(A, k=1), [2])
      assert_equal(diag(A, k=0), [1, 4])
      assert_equal(diag(A, k=-1), [3, 6])
      assert_equal(diag(A, k=-2), [5])
      assert_equal(diag(A, k=-3), [])
  
imports: from numpy.testing import assert_equal, assert_array_equal, assert_array_max_ulp, assert_array_almost_equal, assert_raises, assert_
  from numpy import arange, add, fliplr, flipud, zeros, ones, eye, array, diag, histogram2d, tri, mask_indices, triu_indices, triu_indices_from, tril_indices, tril_indices_from, vander
  import numpy
  import pytest
label_desc: Use image node_id 4 for calling the TestDiag obj's underlying member method code with example usage: obj.test_diag_bounds() without return types