code: def test_write_read_rgba_uint16_png(tmpdir):
      filename = tmpdir / "rgba_uint16.png"
      red = np.arange(0, 65536, 64, dtype=np.uint16).reshape(
          (32, 32, 1)
      )
      img_data = np.concatenate(
          [
              red,
              np.zeros_like(red),
              np.ones_like(red),
              np.ones_like(red),
          ],
          axis=-1,
      )
      file_io.write_png(img_data, 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 5 for calling a global function with example usage: test_write_read_rgba_uint16_png(tmpdir) without return types