travis friendly

tests/test_processing.py

@@ -1,8 +1,5 @@
 # coding: utf-8
 import os
-
-import multiprocessing
-
 import psutil
 import pytest
 
@@ -10,6 +7,8 @@ from synergine2.processing import ProcessManager
 from synergine2.utils import ChunkManager
 from tests import BaseTest
 
+available_cores = len(psutil.Process().cpu_affinity())
+
 
 class MyFakeClass(object):
     def __init__(self, value):
@@ -38,11 +37,11 @@ class TestProcessing(BaseTest):
         result = sum(data)
         return current_pid, MyFakeClass(result)
 
-    @pytest.mark.skipif(len(psutil.Process().cpu_affinity()) < 2, reason="requires 2 or more cpus")
+    @pytest.mark.skipif(available_cores < 2, reason="requires 2 or more cpus")
     def test_parallel_jobs_with_scalar(self):
-        chunk_manager = ChunkManager(2)
+        chunk_manager = ChunkManager(available_cores)
         process_manager = ProcessManager(
-            process_count=2,
+            process_count=available_cores,
             chunk_manager=chunk_manager,
         )
 
@@ -84,11 +83,11 @@ class TestProcessing(BaseTest):
         assert process_id == os.getpid()
         assert final_result == 4950
 
-    @pytest.mark.skipif(len(psutil.Process().cpu_affinity()) < 2, reason="requires 2 or more cpus")
+    @pytest.mark.skipif(available_cores < 2, reason="requires 2 or more cpus")
     def test_parallel_jobs_with_objects(self):
-        chunk_manager = ChunkManager(4)
+        chunk_manager = ChunkManager(available_cores)
         process_manager = ProcessManager(
-            process_count=4,
+            process_count=available_cores,
             chunk_manager=chunk_manager,
         )