code: def fat_cases(align=None):
      # pyformat: disable
      mat = np.array(
          [
              [[1, 2, 3, 4], [5, 6, 7, 8], [9, 1, 2, 3]],
              [[4, 5, 6, 7], [8, 9, 1, 2], [3, 4, 5, 6]],
          ]
      )
      tests = dict()
      # tests[d_lower, d_upper] = (compact_diagonals, padded_diagonals)
      tests[0, 0] = (
          np.array([[1, 6, 2], [4, 9, 5]]),
          np.array(
              [
                  [[1, 0, 0, 0], [0, 6, 0, 0], [0, 0, 2, 0]],
                  [[4, 0, 0, 0], [0, 9, 0, 0], [0, 0, 5, 0]],
              ]
          ),
      )
      tests[2, 2] = (
          np.array([[3, 8], [6, 2]]),
          np.array(
              [
                  [[0, 0, 3, 0], [0, 0, 0, 8], [0, 0, 0, 0]],
                  [[0, 0, 6, 0], [0, 0, 0, 2], [0, 0, 0, 0]],
              ]
          ),
      )
      tests[-2, 0] = (
          np.array(
              [
                  [[1, 6, 2], [5, 1, 0], [9, 0, 0]],
                  [[4, 9, 5], [8, 4, 0], [3, 0, 0]],
              ]
          ),
          np.array(
              [
                  [[1, 0, 0, 0], [5, 6, 0, 0], [9, 1, 2, 0]],
                  [[4, 0, 0, 0], [8, 9, 0, 0], [3, 4, 5, 0]],
              ]
          ),
      )
      tests[-1, 1] = (
          np.array(
              [
                  [[2, 7, 3], [1, 6, 2], [5, 1, 0]],
                  [[5, 1, 6], [4, 9, 5], [8, 4, 0]],
              ]
          ),
          np.array(
              [
                  [[1, 2, 0, 0], [5, 6, 7, 0], [0, 1, 2, 3]],
                  [[4, 5, 0, 0], [8, 9, 1, 0], [0, 4, 5, 6]],
              ]
          ),
      )
      tests[0, 3] = (
          np.array(
              [
                  [[4, 0, 0], [3, 8, 0], [2, 7, 3], [1, 6, 2]],
                  [[7, 0, 0], [6, 2, 0], [5, 1, 6], [4, 9, 5]],
              ]
          ),
          np.array(
              [
                  [[1, 2, 3, 4], [0, 6, 7, 8], [0, 0, 2, 3]],
                  [[4, 5, 6, 7], [0, 9, 1, 2], [0, 0, 5, 6]],
              ]
          ),
      )
      # pyformat: enable
      return (mat, repack_diagonals_in_tests(tests, align))
  
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gen_array_ops
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 6 for calling a global function with example usage: fat_cases(align) and returns: mat,  repack_diagonals_in_tests