Fix tests for python 2

ildus · ildus · commit 8b7d0d1abc4f · 2018-06-25T18:27:59.000+03:00
diff --git a/tests/test_simple.py b/tests/test_simple.py
@@ -7,6 +7,7 @@
 import tempfile
 import testgres
 import time
+import six
 import unittest
 
 import logging.config
@@ -859,7 +860,7 @@ def test_version_management(self):
 
         version = get_pg_version()
         with get_new_node() as node:
-            self.assertTrue(isinstance(version, str))
+            self.assertTrue(isinstance(version, six.string_types))
             self.assertTrue(isinstance(node.version, PgVer))
             self.assertTrue(node.version == version)
 
