run-tests: move all open-coded sys.version_info checks to PYTHON3 (issue4668)
authorAugie Fackler <augie@google.com>
Sun, 17 May 2015 21:18:56 -0400
changeset 25159 138dc8381049
parent 25158 3906b9783cd9
child 25160 fefc72523491
run-tests: move all open-coded sys.version_info checks to PYTHON3 (issue4668) This consolidates all version checking in a single place and helps the code read more obviously.
tests/run-tests.py
--- a/tests/run-tests.py	Sun May 17 21:17:13 2015 -0400
+++ b/tests/run-tests.py	Sun May 17 21:18:56 2015 -0400
@@ -323,7 +323,7 @@
     os.remove(src)
 
 _unified_diff = difflib.unified_diff
-if sys.version_info[0] > 2:
+if PYTHON3:
     import functools
     _unified_diff = functools.partial(difflib.diff_bytes, difflib.unified_diff)
 
@@ -831,7 +831,7 @@
 ]
 
 bchr = chr
-if sys.version_info[0] == 3:
+if PYTHON3:
     bchr = lambda x: bytes([x])
 
 class TTest(Test):
@@ -1118,7 +1118,7 @@
             return True
         if el:
             if el.endswith(b" (esc)\n"):
-                if sys.version_info[0] == 3:
+                if PYTHON3:
                     el = el[:-7].decode('unicode_escape') + '\n'
                     el = el.encode('utf-8')
                 else:
@@ -1278,7 +1278,7 @@
                 pass
             elif self._options.view:
                 v = self._options.view
-                if sys.version_info[0] == 3:
+                if PYTHON3:
                     v = v.encode('utf-8')
                 os.system(b"%s %s %s" %
                           (v, test.refpath, test.errpath))
@@ -1292,7 +1292,7 @@
                 else:
                     self.stream.write('\n')
                     for line in lines:
-                        if sys.version_info[0] > 2:
+                        if PYTHON3:
                             self.stream.flush()
                             self.stream.buffer.write(line)
                             self.stream.buffer.flush()
@@ -1736,7 +1736,7 @@
             # If --with-hg is not specified, we have bytes already,
             # but if it was specified in python3 we get a str, so we
             # have to encode it back into a bytes.
-            if sys.version_info[0] == 3:
+            if PYTHON3:
                 if not isinstance(whg, bytes):
                     whg = whg.encode('utf-8')
             self._bindir = os.path.dirname(os.path.realpath(whg))
@@ -1762,7 +1762,7 @@
 
         fileb = __file__.encode('utf-8')
         runtestdir = os.path.abspath(os.path.dirname(fileb))
-        if sys.version_info[0] == 3:
+        if PYTHON3:
             sepb = os.pathsep.encode('utf-8')
         else:
             sepb = os.pathsep
@@ -1848,7 +1848,7 @@
             failed = False
             warned = False
             kws = self.options.keywords
-            if kws is not None and sys.version_info[0] == 3:
+            if kws is not None and PYTHON3:
                 kws = kws.encode('utf-8')
 
             suite = TestSuite(self._testdir,
@@ -1995,7 +1995,7 @@
         # Run installer in hg root
         script = os.path.realpath(sys.argv[0])
         exe = sys.executable
-        if sys.version_info[0] == 3:
+        if PYTHON3:
             py3 = b'--c2to3'
             compiler = compiler.encode('utf-8')
             script = script.encode('utf-8')
@@ -2039,7 +2039,7 @@
         else:
             f = open(installerrs, 'rb')
             for line in f:
-                if sys.version_info[0] > 2:
+                if PYTHON3:
                     sys.stdout.buffer.write(line)
                 else:
                     sys.stdout.write(line)
@@ -2115,12 +2115,12 @@
 
         cmd = b'%s -c "import mercurial; print (mercurial.__path__[0])"'
         cmd = cmd % PYTHON
-        if sys.version_info[0] > 2:
+        if PYTHON3:
             cmd = cmd.decode('utf-8')
         pipe = os.popen(cmd)
         try:
             self._hgpath = pipe.read().strip()
-            if sys.version_info[0] == 3:
+            if PYTHON3:
                 self._hgpath = self._hgpath.encode('utf-8')
         finally:
             pipe.close()
@@ -2157,7 +2157,7 @@
 
     def _findprogram(self, program):
         """Search PATH for a executable program"""
-        if sys.version_info[0] > 2:
+        if PYTHON3:
             dpb = os.defpath.encode('utf-8')
             sepb = os.pathsep.encode('utf-8')
         else: