Skip to content

Commit 75cd06d

Browse files
committed
Fix irrelevant Windows tests
1 parent ff14653 commit 75cd06d

File tree

1 file changed

+17
-6
lines changed

1 file changed

+17
-6
lines changed

Lib/test/test_pyrepl/test_windows_console.py

+17-6
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@
1111
from unittest import TestCase
1212
from unittest.mock import MagicMock, call
1313

14-
from .support import handle_all_events, code_to_events
14+
from .support import handle_all_events, code_to_events, reader_no_colors
15+
from .support import prepare_reader as default_prepare_reader
1516

1617
try:
1718
from _pyrepl.console import Event, Console
@@ -46,14 +47,22 @@ def console(self, events, **kwargs) -> Console:
4647
setattr(console, key, val)
4748
return console
4849

49-
def handle_events(self, events: Iterable[Event], **kwargs):
50-
return handle_all_events(events, partial(self.console, **kwargs))
50+
def handle_events(
51+
self,
52+
events: Iterable[Event],
53+
prepare_console=None,
54+
prepare_reader=None,
55+
**kwargs,
56+
):
57+
prepare_console = prepare_console or partial(self.console, **kwargs)
58+
prepare_reader = prepare_reader or default_prepare_reader
59+
return handle_all_events(events, prepare_console, prepare_reader)
5160

5261
def handle_events_narrow(self, events):
5362
return self.handle_events(events, width=5)
5463

55-
def handle_events_short(self, events):
56-
return self.handle_events(events, height=1)
64+
def handle_events_short(self, events, **kwargs):
65+
return self.handle_events(events, height=1, **kwargs)
5766

5867
def handle_events_height_3(self, events):
5968
return self.handle_events(events, height=3)
@@ -248,7 +257,9 @@ def test_resize_bigger_on_multiline_function(self):
248257
# fmt: on
249258

250259
events = itertools.chain(code_to_events(code))
251-
reader, console = self.handle_events_short(events)
260+
reader, console = self.handle_events_short(
261+
events, prepare_reader=reader_no_colors
262+
)
252263

253264
console.height = 2
254265
console.getheightwidth = MagicMock(lambda _: (2, 80))

0 commit comments

Comments
 (0)