code: def test_argmax():
      """Test whether the logprob for ```pt.argmax``` is correctly rejected"""
      x = pt.random.normal(0, 1, size=(3,))
      x.name = "x"
      x_max = pt.argmax(x, axis=-1)
      x_max_value = pt.vector("x_max_value")
  
      with pytest.raises(
          RuntimeError,
          match=re.escape("Logprob method not implemented for Argmax"),
      ):
          x_max_logprob = logp(x_max, x_max_value)
  
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 1 for calling a global function with example usage: test_argmax() without return types