• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• loky-master
  • .coveragerc
  • examples
    • README.txt
    • cloudpickle_wrapper.py
    • reuse_executor.py
    • reuseable_executor.py
    • deadlocks
      • deadlock_timeout_shutdown.py
      • deadlock_on_shutdown.py
      • README.txt
      • deadlock_on_failed_pickle.py
      • deadlock_results.py
  • continuous_integration
    • install_coverage_subprocess_pth.py
    • build_test_ext.sh
    • runtests.sh
    • copy_loky.sh
  • CHANGES.md
  • .readthedocs.yml
  • dev-requirements.txt
  • TODO.md
  • README.rst
  • setup.py
  • setup.cfg
  • README.md
  • tests
    • test_loky_module.py
    • test_process_executor_loky.py
    • _executor_mixin.py
    • test_loky_backend.py
    • test_resource_tracker.py
    • test_process_executor_forkserver.py
    • _test_process_executor.py
    • test_cloudpickle_wrapper.py
    • test_futures.py
    • _openmp
      • _fail_fork.py
      • setup.py
      • __init__.py
      • parallel_sum.pyx
      • .gitignore
    • test_synchronize.py
    • test_process_executor_spawn.py
    • test_reusable_executor.py
    • __init__.py
    • utils.py
    • conftest.py
    • test_worker_timeout.py
  • .azure_pipeline.yml
  • loky
    • reusable_executor.py
    • process_executor.py
    • backend
      • spawn.py
      • _posix_wait.py
      • resource_tracker.py
      • _win_reduction.py
      • synchronize.py
      • queues.py
      • managers.py
      • popen_loky_posix.py
      • _win_wait.py
      • compat_win32.py
      • popen_loky_win32.py
      • __init__.py
      • fork_exec.py
      • utils.py
      • compat_posix.py
      • semlock.py
      • process.py
      • compat.py
      • context.py
      • _posix_reduction.py
      • reduction.py
    • cloudpickle_wrapper.py
    • __init__.py
    • _base.py
  • .codecov.yml
  • .gitignore
  • docs
    • Makefile
    • examples.rst
    • README.rst
    • API.rst
    • requirements.txt
    • index.rst
    • conf.py
    • .gitignore
    • _static
      • custom.css
  • LICENSE.txt
  • MANIFEST.in
  • tox.ini

import os
import sys
import shutil
from subprocess import check_output

import pytest

import loky
from loky import cpu_count


def test_version():
    assert hasattr(loky, '__version__'), (
        "There are no __version__ argument on the loky module")


def test_cpu_count():
    cpus = cpu_count()
    assert type(cpus) is int
    assert cpus >= 1


cpu_count_cmd = ("from loky.backend.context import cpu_count;"
                 "print(cpu_count())")


def test_cpu_count_affinity():
    if not hasattr(os, 'sched_getaffinity') or not hasattr(shutil, 'which'):
        pytest.skip()

    taskset_bin = shutil.which('taskset')
    python_bin = shutil.which('python')

    if taskset_bin is None or python_bin is None:
        raise pytest.skip()

    try:
        os.sched_getaffinity(0)
    except NotImplementedError:
        pytest.skip()

    res = check_output([taskset_bin, '-c', '0',
                        python_bin, '-c', cpu_count_cmd])

    assert res.strip().decode('utf-8') == '1'


def test_cpu_count_cfs_limit():
    if sys.platform == "win32":
        pytest.skip()

    if not hasattr(shutil, 'which'):
        pytest.skip()

    docker_bin = shutil.which('docker')
    if docker_bin is None:
        raise pytest.skip()

    loky_path = os.path.abspath(os.path.dirname(loky.__file__))

    # The following will always run using the Python 3.6 docker image.
    # We mount the loky source as /loky inside the container,
    # so it can be imported when running commands under /
    res = check_output([docker_bin, 'run', '--rm', '--cpus', '0.5',
                        '-v', '%s:/loky' % loky_path,
                        'python:3.6',
                        'python', '-c', cpu_count_cmd])

    assert res.strip().decode('utf-8') == '1'