Làm thế nào để kiểm tra trang trí hàm Python?

Question

Tôi đang cố gắng viết các bài kiểm tra đơn vị để đảm bảo tính chính xác của các trang trí khác nhau mà tôi đã viết. Đây là sự khởi đầu của mã tôi đang cố gắng viết:

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() 

Trong khi điều này làm việc cho chiến lược MRU tôi đã nêu ở trên, tôi sẽ cần phải trang trí thêm chức năng fib theo nhiều cách khác nhau. (Nhớ lại rằng bởi vì fib gọi fib, thiết lập fib2 = memoize (fib) không ghi nhớ các giá trị trung gian để nó không hoạt động.) Cách chính xác để kiểm tra các trang trí khác là gì?

Answer 1

Hãy nhìn vào các bài kiểm tra trong thư viện chuẩn cho ví dụ: http://hg.python.org/cpython/file/3.2/Lib/test/test_functools.py#l553

tôi thường thêm một số thiết bị để chức năng được bọc để tôi có thể giám sát các cuộc gọi.

Thay vì ghi nhớ chức năng kiểm tra ở cấp mô-đun, tôi tạo hàm đã ghi nhớ bên trong phép thử sao cho thử nghiệm mới được tạo cho mỗi thử nghiệm và cho từng biến trang trí.

Answer 2

gì về khá phức tạp

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

Bằng cách này bạn có thể làm

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)