नाक

Question

के साथ पाइथन मल्टीप्रोसेसिंग पूल कोड परीक्षण nose के साथ परीक्षण लिखने की कोशिश कर रहा हूं जो मल्टीप्रोसेसिंग का उपयोग करके गणना की गई है।

code/ 
    tests/ 
     tests.py 

tests.py इस तरह दिखता है:

import multiprocessing as mp 


def f(i): 
    return i ** 2 


pool = mp.Pool() 
out = pool.map(f, range(10)) 


def test_pool(): 
    """Really simple test that relies on the output of pool.map. 
    The actual tests are much more complicated, but this is all 
    that is needed to produce the problem.""" 
    ref_out = map(f, range(10)) 
    assert out == ref_out 

if __name__ == '__main__': 
    test_pool() 

code निर्देशिका से चल रहा है, python tests/tests.pyगुजरता

मैं इस निर्देशिका संरचना है।

nosetests tests/tests.pyपूरा करने के लिए विफल रहता है। यह शुरू होता है, लेकिन pool.map पर कॉल के माध्यम से कभी नहीं मिलता है और बस लटकता है।

यह क्यों है और सबसे आसान समाधान क्या है?

Answer 1

समस्या इस तथ्य से संबंधित है कि pool.map को "वैश्विक स्तर" पर बुलाया जाता है। आम तौर पर आप इससे बचना चाहते हैं, क्योंकि इन बयानों को निष्पादित किया जाएगा भले ही आपकी फ़ाइल बस आयात की गई हो।

नाक को import your module आपके परीक्षणों को खोजने में सक्षम होने के लिए है और बाद में उन्हें निष्पादित करने में सक्षम है, इसलिए मेरा मानना ​​है कि समस्या तब होती है जब आयात तंत्र में प्रवेश होता है (मैंने इस व्यवहार के लिए सही कारण खोजने का प्रयास नहीं किया है)

आपको अपने प्रारंभिक कोड को इसके बजाय एक परीक्षण स्थिरता में ले जाना चाहिए; नाक with_setup सजावट के साथ जुड़नार का समर्थन करता है।

import multiprocessing as mp 
from nose import with_setup 

pool = None 
out = None 

def f(i): 
    return i ** 2 

def setup_func(): 
    global pool 
    global out 
    pool = mp.Pool() 
    out = pool.map(f, range(10)) 

@with_setup(setup_func) 
def test_pool(): 
    """Really simple test that relies on the output of pool.map. 
    The actual tests are much more complicated, but this is all 
    that is needed to produce the problem.""" 
    global out 
    ref_out = map(f, range(10)) 
    assert out == ref_out 

if __name__ == '__main__': 
    test_pool() 

निष्पादित:

$ nosetests tests/tests.py 
. 
---------------------------------------------------------------------- 
Ran 1 test in 0.011s 

OK