code: def test_max_discrete(mu, size, value, axis):
      x = pm.Poisson.dist(name="x", mu=mu, size=(size))
      x_max = pt.max(x, axis=axis)
      x_max_value = pt.scalar("x_max_value")
      x_max_logprob = logp(x_max, x_max_value)
  
      test_value = value
  
      n = size
      exp_rv = sp.poisson(mu).cdf(test_value) ** n
      exp_rv_prev = sp.poisson(mu).cdf(test_value - 1) ** n
  
      np.testing.assert_allclose(
          np.log(exp_rv - exp_rv_prev),
          (x_max_logprob.eval({x_max_value: test_value})),
          rtol=1e-06,
      )
  
file_path: pymc/tests/logprob/test_order.py
imports: import re
  import numpy
  import pytensor.tensor
  import pytest
  import scipy.stats
  import pymc
  from pymc import logp
  from pymc.testing import assert_no_rvs
label_desc: Use image node_id 10 for calling a global function with example usage: test_max_discrete(mu, size, value, axis) without return types