X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/a11c56ef66604a9117e3db8c2fa2273c98f88b51..4d0d27f83183b6659a570ae1fc07f95aefe0b23e:/sdk/python/tests/test_pipeline_template.py

diff --git a/sdk/python/tests/test_pipeline_template.py b/sdk/python/tests/test_pipeline_template.py
index bc82271f9c..4c0059aaea 100644
--- a/sdk/python/tests/test_pipeline_template.py
+++ b/sdk/python/tests/test_pipeline_template.py
@@ -1,3 +1,4 @@
+from __future__ import absolute_import
 # usage example:
 #
 # ARVADOS_API_TOKEN=abc ARVADOS_API_HOST=arvados.local python -m unittest discover
@@ -5,11 +6,11 @@
 import unittest
 import arvados
 import apiclient
-import run_test_server
+from . import run_test_server
 
-class PipelineTemplateTest(unittest.TestCase):
-    def setUp(self):
-        run_test_server.run()
+class PipelineTemplateTest(run_test_server.TestCaseWithServers):
+    MAIN_SERVER = {}
+    KEEP_SERVER = {}
 
     def runTest(self):
         run_test_server.authorize_with("admin")
@@ -55,6 +56,3 @@ class PipelineTemplateTest(unittest.TestCase):
             geterror_response = arvados.api('v1').pipeline_templates().get(
                 uuid=pt_uuid
                 ).execute()
-
-    def tearDown(self):
-        run_test_server.stop()