小编R. *_*ard的帖子

测试使用多处理的代码时,nose.main挂起

我有一个存储函数的类和一个尝试使用multiprocessing模块并行运行该函数的方法。我正在使用该nose模块测试该类,并且该nose.main()函数在运行时挂起(什么也没有发生,没有显示任何输出,必须手动告知内核停止并重新启动)。我的目录如下:

Parent/
    test.py
    Code/
        __init__.py
        code.py
    tests/
        test_code.py
Run Code Online (Sandbox Code Playgroud)

code.py 包含:

import multiprocessing as mp
import numpy as np
import itertools

def target(): pass
def target_star(args): return target(*args)

class FuncWrapper():
    def __init__(self, fun):
        self.fun = fun

    def run(self, X):
        try:
            arg_list_objects = []
            arg_list_inputs = []

            for i in range(len(X)):
                arg_list_objects.append(self.fun.im_self)
                arg_list_inputs.append(X[i])

            target.__code__ = self.fun.im_func.__code__

            pool = mp.Pool(processes=mp.cpu_count()-1)

            F = np.array(pool.map(target_star, itertools.izip(arg_list_objects, arg_list_inputs))).reshape(X.shape)

            pool.close()
            pool.join()
        except:
            F = self.fun(X)

        return F
Run Code Online (Sandbox Code Playgroud)

try:块中的大多数代码用于使pool.map函数与类方法一起使用。 …

python nose python-2.7 python-multiprocessing

5
推荐指数
0
解决办法
626
查看次数

标签 统计

nose ×1

python ×1

python-2.7 ×1

python-multiprocessing ×1