code: def test_max_as_distribution(self):
      self.assertIsInstance(self.qout.max_as_distribution, torch.Tensor)
      for b in range(self.batch_size):
          np.testing.assert_almost_equal(
              self.qout.max_as_distribution.detach().cpu().numpy()[b],
              self.atom_probs[
                  b, self.qout.greedy_actions.detach().cpu().numpy()[b]
              ],
              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 3 for calling the TestDistributionalDiscreteActionValue obj's underlying member method code with example usage: obj.test_max_as_distribution() without return types