From d68adb73973d57e01504f8f8778f8b0610c4dd64 Mon Sep 17 00:00:00 2001
From: Carlos Natalino <carlos.natalino@chalmers.se>
Date: Thu, 29 Sep 2022 13:00:56 +0200
Subject: [PATCH] Fixing tests of the dbscanserving.
---
src/dbscanserving/client/DbscanServingClient.py | 3 +--
src/dbscanserving/tests/test_unitary.py | 15 ++++++++++++---
2 files changed, 13 insertions(+), 5 deletions(-)
diff --git a/src/dbscanserving/client/DbscanServingClient.py b/src/dbscanserving/client/DbscanServingClient.py
index 7794115c5..be653567a 100644
--- a/src/dbscanserving/client/DbscanServingClient.py
+++ b/src/dbscanserving/client/DbscanServingClient.py
@@ -66,8 +66,7 @@ class DbscanServingClient:
response: DetectionResponse = self.stub.Detect(request)
LOGGER.debug(
"Detect result with {} cluster indices [{}]".format(
- len(response.cluster_indices),
- Counter(response.cluster_indices)
+ len(response.cluster_indices), Counter(response.cluster_indices)
)
)
return response
diff --git a/src/dbscanserving/tests/test_unitary.py b/src/dbscanserving/tests/test_unitary.py
index 5498b184f..9de9b9c91 100644
--- a/src/dbscanserving/tests/test_unitary.py
+++ b/src/dbscanserving/tests/test_unitary.py
@@ -13,9 +13,10 @@
# limitations under the License.
import logging
+import os
import random
+from unittest.mock import patch
-import numpy
import pytest
from common.proto.dbscanserving_pb2 import DetectionRequest, DetectionResponse, Sample
from dbscanserving.client.DbscanServingClient import DbscanServingClient
@@ -40,8 +41,16 @@ def dbscanserving_service():
@pytest.fixture(scope="session")
def dbscanserving_client():
- _client = DbscanServingClient(host="127.0.0.1", port=port)
- yield _client
+ with patch.dict(
+ os.environ,
+ {
+ "DBSCANSERVINGSERVICE_SERVICE_HOST": "127.0.0.1",
+ "DBSCANSERVINGSERVICE_SERVICE_PORT_GRPC": str(port),
+ },
+ clear=True,
+ ):
+ _client = DbscanServingClient()
+ yield _client
_client.close()
--
GitLab