|
@@ -2,6 +2,8 @@
|
2
|
2
|
import os
|
3
|
3
|
|
4
|
4
|
import multiprocessing
|
|
5
|
+
|
|
6
|
+import psutil
|
5
|
7
|
import pytest
|
6
|
8
|
|
7
|
9
|
from synergine2.processing import ProcessManager
|
|
@@ -36,7 +38,7 @@ class TestProcessing(BaseTest):
|
36
|
38
|
result = sum(data)
|
37
|
39
|
return current_pid, MyFakeClass(result)
|
38
|
40
|
|
39
|
|
- @pytest.mark.skipif(multiprocessing.cpu_count() < 2, reason="requires 2 or more cpus")
|
|
41
|
+ @pytest.mark.skipif(len(psutil.Process().cpu_affinity()) < 2, reason="requires 2 or more cpus")
|
40
|
42
|
def test_parallel_jobs_with_scalar(self):
|
41
|
43
|
chunk_manager = ChunkManager(2)
|
42
|
44
|
process_manager = ProcessManager(
|
|
@@ -82,6 +84,7 @@ class TestProcessing(BaseTest):
|
82
|
84
|
assert process_id == os.getpid()
|
83
|
85
|
assert final_result == 4950
|
84
|
86
|
|
|
87
|
+ @pytest.mark.skipif(len(psutil.Process().cpu_affinity()) < 2, reason="requires 2 or more cpus")
|
85
|
88
|
def test_parallel_jobs_with_objects(self):
|
86
|
89
|
chunk_manager = ChunkManager(4)
|
87
|
90
|
process_manager = ProcessManager(
|