Come testare i decoratori di funzioni Python?

Question

Sto provando a scrivere test unitari per garantire la correttezza dei vari decoratori che ho scritto. Ecco l'inizio del codice che sto cercando di scrivere:

import unittest 

from memoizer import Memoizer 
from strategies.mru import MRU 


@Memoizer(strategy=MRU(maxsize=10)) 
def fib(x): 
    if x < 2: 
    return 1 
    else: 
    return fib(x-1) + fib(x-2) 


class TestMemoizer(unittest.TestCase): 

    def test_simple(self): 
    self.assertEqual(fib(0), 1) 
    self.assertEqual(fib(1), 1) 
    self.assertEqual(fib(10), 89) 


if __name__ == '__main__': 
    unittest.main() 

Anche se questo funziona decentemente per la strategia MRU che ho sopra, ho intenzione di scrivere ulteriori strategie, nel qual caso avrò bisogno di decorare con la funzione fib in diversi modi. (Ricorda che poiché fib chiama fib, l'impostazione fib2 = memoize (fib) non memorizza i valori intermedi in modo che non funzioni.) Qual è il modo corretto di testare gli altri decoratori?

Answer 1

Date un'occhiata ai test nella libreria standard per gli esempi: http://hg.python.org/cpython/file/3.2/Lib/test/test_functools.py#l553

Io di solito aggiungere un po ' strumentazione alla funzione che è stata spostata in modo da poter monitorare le chiamate.

Invece di memoizzare la funzione di test a livello di modulo, creo la funzione memoized all'interno del test in modo che ne venga creata una nuova per ciascun test e per ogni variante di decoratore.

Answer 2

Che dire del piuttosto complicato

def mkfib(strategy): 
    @Memoizer(strategy=strategy) 
    def fib(x): 
     if x < 2: 
     return 1 
     else: 
     return fib(x-1) + fib(x-2) 
    return fib 

In questo modo si potrebbe fare

fib1 = mkfib(MRU(maxsize=10)) 
self.assertEqual(fib1(0), 1) 
self.assertEqual(fib1(1), 1) 

fib2 = mkfib(MRU(maxsize=10)) # produces another cache 
self.assertEqual(fib2(0), 1) 
self.assertEqual(fib2(1), 1)