diff --git a/src/tests/test_process.py b/src/tests/test_process.py index efe4adcb..8e064d00 100644 --- a/src/tests/test_process.py +++ b/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):