Merge pull request #980 from python-gitlab/refactor/cleanup-upgrade

Refactor/cleanup upgrade

1 files changed, 3 insertions, 4 deletions

diff --git a/gitlab/tests/test_cli.py b/gitlab/tests/test_cli.py
index 04a1961..4820103 100644
--- a/gitlab/tests/test_cli.py
+++ b/gitlab/tests/test_cli.py
@@ -20,6 +20,7 @@ import argparse
 import os
 import tempfile
 import unittest
+import io
 
 try:
     from contextlib import redirect_stderr  # noqa: H302
@@ -34,8 +35,6 @@ except ImportError:
         sys.stderr = old_target
 
 
-import six
-
 from gitlab import cli
 import gitlab.v4.cli
 
@@ -56,7 +55,7 @@ class TestCLI(unittest.TestCase):
         self.assertEqual("class", cli.cls_to_what(Class))
 
     def test_die(self):
-        fl = six.StringIO()
+        fl = io.StringIO()
         with redirect_stderr(fl):
             with self.assertRaises(SystemExit) as test:
                 cli.die("foobar")
@@ -83,7 +82,7 @@ class TestCLI(unittest.TestCase):
         self.assertEqual(ret, "content")
         os.unlink(temp_path)
 
-        fl = six.StringIO()
+        fl = io.StringIO()
         with redirect_stderr(fl):
             with self.assertRaises(SystemExit) as exc:
                 cli._parse_value("@/thisfileprobablydoesntexist")