code: def test_additional(self):
      for p in TEST_NDARRAYS:
          out = AsDiscrete(argmax=True, dim=1, keepdim=False)(
              p([[[0.0, 1.0]], [[2.0, 3.0]]])
          )
          assert_allclose(
              out, p([[0.0, 0.0], [0.0, 0.0]]), type_test=False
          )
  
file_path: MONAI/tests/test_as_discrete.py
imports: from __future__ import annotations
  import unittest
  from parameterized import parameterized
  from monai.transforms import AsDiscrete
  from tests.utils import TEST_NDARRAYS, assert_allclose
label_desc: Use image node_id 2 for calling the TestAsDiscrete obj's underlying member method code with example usage: obj.test_additional() without return types