From 14437dc1e4db3fd3165a96f8ccf0f4373d67eb67 Mon Sep 17 00:00:00 2001 From: Iman Tabrizian Date: Fri, 28 Jul 2023 16:54:07 -0400 Subject: [PATCH] Improve asyncio testing (#6122) --- qa/L0_grpc/python_grpc_aio_test.py | 5 ++++- qa/L0_grpc/test.sh | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/qa/L0_grpc/python_grpc_aio_test.py b/qa/L0_grpc/python_grpc_aio_test.py index f5b3a8f958..f342f19ad5 100755 --- 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 @@ -56,6 +56,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 index 90d34a8738..cf43e324a2 100755 --- a/qa/L0_grpc/test.sh +++ b/qa/L0_grpc/test.sh @@ -653,3 +653,4 @@ else fi exit $RET +