code: def test_max(self):
      self.assertIsInstance(self.qout.max, torch.Tensor)
      np.testing.assert_almost_equal(
          self.qout.max.detach().cpu().numpy(),
          self.q_values.max(axis=1),
          decimal=5,
      )
  
imports: import unittest
  import numpy
  import pytest
  import torch
  from pfrl import action_value
  from pfrl.testing import torch_assert_allclose
label_desc: Use image node_id 2 for calling the TestDistributionalDiscreteActionValue obj's underlying member method code with example usage: obj.test_max() without return types