about summary refs log tree commit diff
path: root/pkgs/by-name/nb/nbmerge/pytest-compatibility.patch
blob: 81ea4c4218f83e0a6fa47303437907075d85c016 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
diff --git a/tests/test_merge.py b/tests/test_merge.py
index 06ee559..9d7b1af 100644
--- a/tests/test_merge.py
+++ b/tests/test_merge.py
@@ -1,3 +1,4 @@
+import io
 import os
 import sys
 import unittest
@@ -21,8 +22,7 @@ def file_names_from(file_paths):
 
 class TestMerge(unittest.TestCase):
     def setUp(self):
-        if not hasattr(sys.stdout, "getvalue"):
-            self.fail("need to run in buffered mode")
+        sys.stdout = io.StringIO()
 
     def _validate_merged_three(self, merged):
         self.assertEqual(len(merged.cells), 6)