पायथन यूनिट परीक्षण के लिए नकली कक्षा विधि कैसे आपूर्ति करें?

Question

मान लें कि मेरे पास इस तरह की कक्षा है।

class SomeProductionProcess(CustomCachedSingleTon): 

     def loaddata(): 
      """ 
      Uses an iterator over a large file in Production for the Data pipeline. 
      """ 

      pass 

अब परीक्षण समय मैं loaddata() विधि के अंदर तर्क बदलना चाहते हैं। यह एक साधारण कस्टम तर्क होगा जो बड़े डेटा को संसाधित नहीं करता है।

हम पाइथन मॉक यूनिटटेस्ट फ्रेमवर्क का उपयोग करके परीक्षण समय पर loaddata() के कस्टम कार्यान्वयन की आपूर्ति कैसे करते हैं?

Answer 1

चलें कहते हैं कि तुम awesome.py और उस में नाम के एक मॉड्यूल है, तो आप के लिए किया था:

import time 

class SomeProductionProcess(CustomCachedSingleTon): 

    def loaddata(self): 
     time.sleep(30) # simulating a long running process 
     return 2 

फिर अपने unittest जहां नकली loaddata ऐसा दिखाई दे सकता:

import unittest 

import awesome # your application module 


class TestSomeProductionProcess(unittest.TestCase): 
    """Example of direct monkey patching""" 

    def test_loaddata(self): 
     some_prod_proc = awesome.SomeProductionProcess() 
     some_prod_proc.loaddata = lambda x: 2 # will return 2 every time called 
     output = some_prod_proc.loaddata() 
     expected = 2 

     self.assertEqual(output, expected) 

या यह इस तरह दिख सकता है:

import unittest 
from mock import patch 

import awesome # your application module 

class TestSomeProductionProcess(unittest.TestCase): 
    """Example of using the mock.patch function""" 

    @patch.object(awesome.SomeProductionProcess, 'loaddata') 
    def test_loaddata(self, fake_loaddata): 
     fake_loaddata.return_value = 2 
     some_prod_proc = awesome.SomeProductionProcess() 

     output = some_prod_proc.loaddata() 
     expected = 2 

     self.assertEqual(output, expected) 

अब जब आप अपना परीक्षण चलाते हैं, loaddata उन परीक्षण मामलों के लिए 30 सेकंड नहीं लेते हैं।

Answer 2

संरचित वापसी_वर्त के साथ कक्षा विधि को आसानी से नकल करने के लिए, unittest.mock.Mock का उपयोग कर सकते हैं।

from unittest.mock import Mock 

mockObject = SomeProductionProcess 
mockObject.loaddata = Mock(return_value=True) 

संपादित करें:

के बाद से आप एक कस्टम कार्यान्वयन के साथ विधि बाहर नकली करना चाहते हैं, तो आप सिर्फ एक कस्टम नकली विधि वस्तु और बाहर मूल विधि स्वैप परीक्षण कार्यावधि में बना सकते हैं।

def custom_method(*args, **kwargs): 
    # do custom implementation 

SomeProductionProcess.loaddata = custom_method 

Answer 3

यहाँ नकली

import mock 


def new_loaddata(cls, *args, **kwargs): 
    # Your custom testing override 
    return 1 


def test_SomeProductionProcess(): 
    with mock.patch.object(SomeProductionProcess, 'loaddata', new=new_loaddata): 
     obj = SomeProductionProcess() 
     obj.loaddata() # This will call your mock method 

मैं unittest मॉड्यूल के बजाय pytest का उपयोग कर यदि आप कर रहे हैं की सलाह देते हैं का उपयोग कर यह करने के लिए एक आसान तरीका है। यह आपके टेस्ट कोड को बहुत क्लीनर बनाता है और unittest.TestCase -स्टाइल परीक्षणों के साथ आपको प्राप्त होने वाले बॉयलरप्लेट को बहुत कम करता है।