code: def test_argmax_1d(self):
      t = np.array([1, 2, 3, 4, 5, 6, 7, 8]).astype(float)
      expected = tf.argmax(t)
  
      with tfe.protocol.SecureNN() as prot:
          out_tfe = prot.argmax(
              prot.define_private_variable(tf.constant(t))
          )
          actual = out_tfe.reveal().to_native()
  
      np.testing.assert_array_equal(actual, expected)
  
imports: import random
  import unittest
  import numpy
  import pytest
  import tensorflow
  import tf_encrypted
  from tf_encrypted.protocol.pond import PondPrivateTensor
  from tf_encrypted.protocol.pond import PondPublicTensor
  from tf_encrypted.protocol.securenn.securenn import _private_compare
label_desc: Use image node_id 1 for calling the TestArgMax obj's underlying member method code with example usage: obj.test_argmax_1d() without return types