为unittesting捕获默认的tqdm输出

Question

如何编写单元测试以确保TQM进度条确实已被调用？另外，当我没有指定filekwarg（在这种情况下，它应该默认为sys.stderr）时，如何捕捉进度条的输出？在

我尝试使用contextlib.redirect_stderr来捕获输出，但是没有成功（请参见下面的详细信息）。在

这是我想测试的。我有一个方法，有时根据self.use_progress_bar使用进度条。对象将被pickle以进行多处理，因此我不能让它拥有进度条。在

def awesome_method(self):
    if self.use_progress_bar:
        progress_bar = tqdm.tqdm(total=self.match_generator.size,
                                 desc="Playing matches")

    if self.filename is not None:
        file = open(self.filename, 'w')
        writer = csv.writer(file, lineterminator='\n')

    for chunk in chunks:
        results = self._play_matches(chunk)
        self._write_interactions(results, writer=writer)

        if self.use_progress_bar:
            progress_bar.update(1)

我想写一个单元测试来确认进度条在那里。我试过这样的方法：

^{pr2}$

但是err是空的。问题似乎是，如果不使用文件实例化tqdm，它默认为系统标准，但不是上下文管理器中的那个。当我试着运行这个：

def action():
err = io.StringIO()
with redirect_stderr(err):
    pbar = tqdm.tqdm(range(5))
    print('pbar: ', pbar.fp)
    print('err: ',err)
    print('stderr:', sys.stderr)
    pbar.update()
    pbar.update()
return err


if __name__ == '__main__':
    err = action()
    print('\nstart\n', err.getvalue(), '\nstop\n')
    print(sys.stderr)

这是输出（减去我无法捕获的pbar输出）：

pbar:  <colorama.ansitowin32.StreamWrapper object at 0x0000000002A18D68>
err:  <_io.StringIO object at 0x0000000002A8B318>
stderr: <_io.StringIO object at 0x0000000002A8B318>

start

stop

<colorama.ansitowin32.StreamWrapper object at 0x0000000002A18D68>

因此，在上下文管理器（进度条被实例化）中，sys.stderr指向err，但是{}的tqdm默认值仍然是获取实际的sys.stderr，即使它在上下文管理器中。为什么？在不设置tqdm对象上的file=的情况下，如何测试进度条？在