code: def test_wide_resnet_general(self):
      x = objax.random.normal((4, 3, 128, 128))
      model = WideResNetGeneral(
          nin=3, nclass=10, blocks_per_group=[4, 4, 4, 4], width=2
      )
      # run in eval mode
      y_eval = model(x, training=False)
      self.assertEqual(y_eval.shape, (4, 10))
      # run in train mode
      y_eval = model(x, training=True)
      self.assertEqual(y_eval.shape, (4, 10))
  
imports: import unittest
  import objax
  from objax.zoo.wide_resnet import WideResNet, WideResNetGeneral
label_desc: Use image node_id 1 for calling the TestWideResNetGeneral obj's underlying member method code with example usage: obj.test_wide_resnet_general() without return types