From 5b8acf7c547fb20b9f687c5d641e214c3e54b11e Mon Sep 17 00:00:00 2001 From: hanhxiao Date: Tue, 13 Aug 2019 16:01:01 +0800 Subject: [PATCH] fix(test): fix routes assert in tests --- tests/test_image_preprocessor.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_image_preprocessor.py b/tests/test_image_preprocessor.py index 9c898dcd..3b735898 100644 --- a/tests/test_image_preprocessor.py +++ b/tests/test_image_preprocessor.py @@ -112,7 +112,7 @@ def test_unary_preprocessor_service_realdata(self): msg.request.index.CopyFrom(req.index) client.send_message(msg) r = client.recv_message() - self.assertEqual(r.envelope.routes[0].service, 'PreprocessorService:BaseUnaryPreprocessor') + self.assertEqual(r.envelope.routes[0].service, 'BaseUnaryPreprocessor') for d in r.request.index.docs: self.assertEqual(len(d.chunks), 1) self.assertEqual(len(blob2array(d.chunks[0].blob).shape), 3) @@ -135,7 +135,7 @@ def test_resize_preprocessor_service_realdata(self): msg.request.index.CopyFrom(req.index) client.send_message(msg) r = client.recv_message() - self.assertEqual(r.envelope.routes[0].service, 'PreprocessorService:PipelinePreprocessor') + self.assertEqual(r.envelope.routes[0].service, 'PipelinePreprocessor') for d in r.request.index.docs: self.assertEqual(len(d.chunks), 1) self.assertEqual(len(blob2array(d.chunks[0].blob).shape), 3) @@ -161,7 +161,7 @@ def test_slidingwindow_preprocessor_service_realdata(self): msg.request.index.CopyFrom(req.index) client.send_message(msg) r = client.recv_message() - self.assertEqual(r.envelope.routes[0].service, 'PreprocessorService:VanillaSlidingPreprocessor') + self.assertEqual(r.envelope.routes[0].service, 'VanillaSlidingPreprocessor') for d in r.request.index.docs: self.assertEqual(len(blob2array(d.chunks[0].blob).shape), 3) self.assertEqual(blob2array(d.chunks[0].blob).shape[-1], 3) @@ -186,7 +186,7 @@ def test_segmentation_preprocessor_service_realdata(self): msg.request.index.CopyFrom(req.index) client.send_message(msg) r = client.recv_message() - self.assertEqual(r.envelope.routes[0].service, 'PreprocessorService:SegmentPreprocessor') + self.assertEqual(r.envelope.routes[0].service, 'SegmentPreprocessor') for d in r.request.index.docs: self.assertEqual(len(blob2array(d.chunks[0].blob).shape), 3) self.assertEqual(blob2array(d.chunks[0].blob).shape[-1], 3)