code: def test_diagflat_execution(setup):
      a = diagflat([[1, 2], [3, 4]], chunk_size=1)
  
      res = a.execute().fetch()
      expected = np.diagflat([[1, 2], [3, 4]])
      np.testing.assert_equal(res, expected)
  
      d = tensor([[1, 2], [3, 4]], chunk_size=1)
      a = diagflat(d)
  
      res = a.execute().fetch()
      expected = np.diagflat([[1, 2], [3, 4]])
      np.testing.assert_equal(res, expected)
  
      a = diagflat([1, 2], 1, chunk_size=1)
  
      res = a.execute().fetch()
      expected = np.diagflat([1, 2], 1)
      np.testing.assert_equal(res, expected)
  
      d = tensor([[1, 2]], chunk_size=1)
      a = diagflat(d, 1)
  
      res = a.execute().fetch()
      expected = np.diagflat([1, 2], 1)
      np.testing.assert_equal(res, expected)
  
imports: import tempfile
  import shutil
  import os
  import time
  import numpy
  import pandas
  import pytest
  from ....None import tensor
  from ....None import dataframe
  from ....lib.sparse import SparseNDArray
  from ....tests.core import require_cupy
  from ....utils import lazy_import
  from ...lib import nd_grid
  from ..None import tensor, ones_like, zeros, zeros_like, full, full_like, arange, empty, empty_like, diag, diagflat, eye, linspace, meshgrid, indices, triu, tril, from_dataframe, fromtiledb, fromhdf5, fromzarr
label_desc: Use image node_id 8 for calling a global function with example usage: test_diagflat_execution(setup) without return types