Improve asyncio testing

triton-inference-server · Jul 28, 2023 · 03e000d · 03e000d
1 parent a8f122d
commit 03e000d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/qa/L0_grpc/python_grpc_aio_test.py b/qa/L0_grpc/python_grpc_aio_test.py
@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-# Copyright 2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
@@ -59,6 +59,9 @@ async def test_get_server_metadata(self):
         ret = await self._triton_client.get_server_metadata()
         self.assertEqual(ret.name, "triton")
 
+        ret = await self._triton_client.get_server_metadata(as_json=True)
+        self.assertEqual(ret['name'], "triton")
+
     async def test_get_model_metadata(self):
         ret = await self._triton_client.get_model_metadata("simple")
         self.assertEqual(ret.name, "simple")

diff --git a/qa/L0_grpc/test.sh b/qa/L0_grpc/test.sh
@@ -653,3 +653,4 @@ else
 fi
 
 exit $RET
+