class_docstr: Unit test for TritonClient class.
class_docstr_tok: ["Unit","test","for","TritonClient","class","."]
code: def test_call(self):
      in_1 = torch.arange(4)
      in_2 = torch.arange(4)
      tmodel = TritonClient(
          model_name="normal",
          input_names=["in_a", "in_b"],
          output_names=["out_a", "out_b"],
      )
      res = tmodel(in_1, in_2)
      for i, j in zip(res, [in_1, in_2]):
          self.assertTrue(i.equal(j))
  
      res = tmodel(in_1, in_b=in_2)
      for i, j in zip(res, [in_1, in_2]):
          self.assertTrue(i.equal(j))
  
      res = tmodel(in_1=in_1, in_b=in_2)
      for i, j in zip(res, [in_1, in_2]):
          self.assertTrue(i.equal(j))
  
file_path: towhee/tests/unittests/serve/triton/test_triton_client.py
imports: import unittest
  import torch
  from towhee.serve.triton.triton_client import TritonClient
  from towhee.serve.triton.bls.python_backend_wrapper import pb_utils
label_desc: Use image node_id 2 for calling the TestTritonClient obj's underlying member method code with example usage: obj.test_call() without return types