code: def test_nchw():
      test_utils.layer_test(
          Maxout,
          kwargs={"num_units": 4, "axis": 1},
          input_shape=(2, 20, 3, 6),
      )
  
      test_utils.layer_test(
          Maxout,
          kwargs={"num_units": 4, "axis": -3},
          input_shape=(2, 20, 3, 6),
      )
  
file_path: addons/tensorflow_addons/layers/tests/maxout_test.py
imports: import pytest
  import numpy
  from tensorflow_addons.layers.maxout import Maxout
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 2 for calling a global function with example usage: test_nchw() without return types