code: def test_write_read_grayscale_uint16_png(tmpdir):
      filename = tmpdir / "grayscale_uint16.png"
      img_data = np.arange(0, 65536, 64, dtype=np.uint16).reshape(
          (32, 32, 1)
      )
      for dtype in [np.uint16, np.uint32, np.uint64]:
          file_io.write_png(img_data.astype(dtype), filename)
          img_data_recovered = file_io.read_png(filename)
          np.testing.assert_array_equal(img_data, img_data_recovered)
  
file_path: kubric/test/test_fileio.py
imports: import numpy
  import pytest
  from kubric import file_io
label_desc: Use image node_id 3 for calling a global function with example usage: test_write_read_grayscale_uint16_png(tmpdir) without return types