py2 compatibility

trbs · Aug 27, 2019 · fb10332 · fb10332
1 parent ce9a6e1
commit fb10332
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_pid.py b/tests/test_pid.py
@@ -175,7 +175,8 @@ def test_pid_already_locked_multi_process():
     pass
 '''
         result = run(['python', '-c', s])
-        assert result.returncode == 1
+        returncode = result if isinstance(result, int) else result.returncode
+        assert returncode == 1
         assert os.path.exists(_pid.filename)
     assert not os.path.exists(_pid.filename)
 
@@ -189,7 +190,8 @@ def test_pid_two_locks_multi_process():
 assert not os.path.exists(_pid.filename)
 '''
         result = run(['python', '-c', s])
-        assert result.returncode == 0
+        returncode = result if isinstance(result, int) else result.returncode
+        assert returncode == 0
         assert os.path.exists(_pid.filename)
     assert not os.path.exists(_pid.filename)