Đơn vị thử nghiệm một hàm trả về một đối tượng phát

Question

Tiêu đề khá nhiều tiền nó lên: Tôi cố gắng sử dụng assertEqual để kiểm tra một hàm trả về một đối tượng phát điện, nhưng có kết quả trong:

AssertionError: generator object genexpr> at 0x2e70c80> != (1,2,...)

Có cách thông minh để làm điều này khác hơn là lặp qua chức năng và chạy assertEqual cho mỗi đầu ra riêng lẻ?

Answer 1

assertEqual(tuple(generator_object), (1, 2, ...)) 

nếu nó là một máy phát vô hạn, hoặc bạn chỉ muốn nhìn vào n kết quả đầu tiên đối với một số lý do, bạn có thể combint này với itertools.islice

assertEqual(tuple(islice(generator_object, n)), (1, 2, ...)) 

Answer 2

Sử dụng next() trên đối tượng phát điện.

assertEqual(next(generator_object), (1, 2, ...))