diff --git a/warden_server/test_warden_server.py b/warden_server/test_warden_server.py index c569230ae9e1a4953b37d6fbdfb7613a579af801..bee3433cf9627751788fb6c779914846b5133398 100755 --- a/warden_server/test_warden_server.py +++ b/warden_server/test_warden_server.py @@ -8,11 +8,15 @@ import sys import warnings from os import path from copy import deepcopy -import unittest2 as unittest import MySQLdb as my from warden_server import build_server import warden_server +if sys.version_info >= (3, 10): + import unittest +else: + import unittest2 + if sys.version_info[0] >= 3: from io import StringIO else: