projects
/
arvados.git
/ blobdiff
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
14012: Rephrases error message.
[arvados.git]
/
sdk
/
python
/
tests
/
test_arv_run.py
diff --git
a/sdk/python/tests/test_arv_run.py
b/sdk/python/tests/test_arv_run.py
index 3d04d272286240cec2ec2d9103a10d6dbaddbda9..1afc1208b49c291f8ba885cf38a2f2211f44d835 100644
(file)
--- a/
sdk/python/tests/test_arv_run.py
+++ b/
sdk/python/tests/test_arv_run.py
@@
-1,16
+1,17
@@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
+# Copyright (C) The Arvados Authors. All rights reserved.
+#
+# SPDX-License-Identifier: Apache-2.0
-import io
+from __future__ import absolute_import
import os
import sys
import tempfile
import unittest
import arvados.commands.run as arv_run
import os
import sys
import tempfile
import unittest
import arvados.commands.run as arv_run
-import arvados_testutil as tutil
+
from .
import arvados_testutil as tutil
-class ArvRunTestCase(unittest.TestCase):
+class ArvRunTestCase(unittest.TestCase
, tutil.VersionChecker
):
def run_arv_run(self, args):
sys.argv = ['arv-run'] + args
return arv_run.main()
def run_arv_run(self, args):
sys.argv = ['arv-run'] + args
return arv_run.main()
@@
-20,10
+21,8
@@
class ArvRunTestCase(unittest.TestCase):
self.run_arv_run(['-x=unknown'])
def test_version_argument(self):
self.run_arv_run(['-x=unknown'])
def test_version_argument(self):
- err = io.BytesIO()
- out = io.BytesIO()
- with tutil.redirected_streams(stdout=out, stderr=err):
+ with tutil.redirected_streams(
+ stdout=tutil.StringIO, stderr=tutil.StringIO) as (out, err):
with self.assertRaises(SystemExit):
self.run_arv_run(['--version'])
with self.assertRaises(SystemExit):
self.run_arv_run(['--version'])
- self.assertEqual(out.getvalue(), '')
- self.assertRegexpMatches(err.getvalue(), "[0-9]+\.[0-9]+\.[0-9]+")
+ self.assertVersionOutput(out, err)