code: def test_reduction_shape(self):
      be = self.be
      x0 = be.array(np.array([[1, 2], [4, 5]]), name="x0")
  
      f = be.sum(x0, axis=0)
      assert f.shape == (1, 2)
  
      f = be.sum(x0, axis=1)
      assert f.shape == (2, 1)
  
      f = be.sum(x0)
      assert f.shape == (1, 1)
  
imports: from builtins import zip
  import numpy
  import pprint
  import pytest
  from neon import NervanaObject
  from neon.backends.autodiff import Autodiff
  from utils import allclose_with_out
label_desc: Use image node_id 7 for calling the TestAutodiff obj's underlying member method code with example usage: obj.test_reduction_shape() without return types