Merge pull request #98 from HansBug/fix/output

dev(hansbug): fix bug for tqdm

hbutils/testing/capture/output.py

@@ -131,11 +131,13 @@ def capture_output(mem: bool = False) -> ContextManager[OutputCaptureResult]:
 
 
 @contextmanager
-def disable_output() -> ContextManager[OutputCaptureResult]:
+def disable_output(encoding='utf-8') -> ContextManager[OutputCaptureResult]:
     """
     Overview:
         Disable all the output to ``sys.stdout`` and ``sys.stderr`` in this ``with`` block.
 
+    :param encoding: Encoding of null file, default is ``utf-8``.
+
     Examples::
         >>> import sys
         >>> from hbutils.testing import disable_output
@@ -144,6 +146,6 @@ def disable_output() -> ContextManager[OutputCaptureResult]:
         ...     print('this is stdout')
         ...     print('this is stderr', file=sys.stderr)
     """
-    with open(os.devnull, 'w') as sout:
+    with open(os.devnull, 'w', encoding=encoding) as sout:
         with redirect_stdout(sout), redirect_stderr(sout):
             yield

requirements-test.txt

@@ -13,4 +13,5 @@ numpy>=1.20; implementation_name != 'pypy' or platform_system != 'Windows' or py
 torch>=1.1.0; python_version < '3.11' and implementation_name != 'pypy'
 faker; python_version > '3.7'
 requests>=2.20
-testtools>=2
+testtools>=2
+tqdm

test/testing/capture/test_output.py

@@ -3,6 +3,7 @@
 from textwrap import dedent
 
 import pytest
+from tqdm.auto import tqdm
 
 from hbutils.testing import capture_output, disable_output, isolated_directory
 
@@ -66,3 +67,13 @@ def test_disable_output(self):
             This is stderr 2rd line.
             This is stderr 3rd line.
         """).strip()
+
+    def test_disable_output_with_tqdm(self):
+        with capture_output() as r:
+            with disable_output():
+                f = tqdm(total=10)
+                f.set_description('你好，这个是中文')
+                f.update(5)
+
+        assert not r.stdout.strip()
+        assert not r.stderr.strip()