Skip to content
Permalink
Browse files

ci(torchvision): install torchvision dependency to enable tests

  • Loading branch information...
hanxiao committed Jul 25, 2019
1 parent 7fcd50d commit 83276f90a6544a279ac2c60d5caca029d1d67f2e
Showing with 3 additions and 12 deletions.
  1. +3 −12 tests/test_preprocessor.py
@@ -16,18 +16,12 @@ def setUp(self):
self.dirname = os.path.dirname(__file__)

def test_preprocessor_service_empty(self):
args = set_preprocessor_service_parser().parse_args([
'--port_in', str(YamlComposer._get_random_port()),
'--port_out', str(YamlComposer._get_random_port())
])
args = set_preprocessor_service_parser().parse_args([])
with PreprocessorService(args):
pass

def test_preprocessor_service_echo(self):
args = set_preprocessor_service_parser().parse_args([
'--port_in', str(YamlComposer._get_random_port()),
'--port_out', str(YamlComposer._get_random_port())
])
args = set_preprocessor_service_parser().parse_args([])
c_args = _set_client_parser().parse_args([
'--port_in', str(args.port_out),
'--port_out', str(args.port_in)
@@ -44,10 +38,7 @@ def test_preprocessor_service_echo(self):
print(r)

def test_preprocessor_service_realdata(self):
args = set_preprocessor_service_parser().parse_args([
'--port_in', str(YamlComposer._get_random_port()),
'--port_out', str(YamlComposer._get_random_port())
])
args = set_preprocessor_service_parser().parse_args([])
c_args = _set_client_parser().parse_args([
'--port_in', str(args.port_out),
'--port_out', str(args.port_in)

0 comments on commit 83276f9

Please sign in to comment.
You can’t perform that action at this time.