src/tests/test_process.py

@@ -9,11 +9,11 @@ import sys
 import tempfile
 import time
 import unittest
-
 import psutil
 
 from .common import cleanup, put_signal_file
 
+PY3 = sys.version_info[0] >= 3
 
 class TestProcessProto(unittest.TestCase):
     """Test case implementing common logic for external testing:
@@ -191,7 +191,10 @@ class TestProcess(TestProcessProto):
     """A test case for pybitmessage process"""
     def test_process_name(self):
         """Check PyBitmessage process name"""
-        self.assertEqual(self.process.name(), 'pybitmessage')
+        if PY3:
+            self.assertEqual(self.process.name(), 'pybitmessage')
+        else:
+            self.assertEqual(self.process.name(), 'PyBitmessage')
 
     @unittest.skipIf(psutil.version_info < (4, 0), 'psutil is too old')
     def test_home(self):