Skip to content

Commit 31573fe

Browse files
committed
Properly close all streams
1 parent e59cf64 commit 31573fe

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

test/test_prototype_datasets_builtin.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -122,12 +122,15 @@ def log_session_streams(self):
122122
@parametrize_dataset_mocks(DATASET_MOCKS)
123123
def test_stream_closing(self, log_session_streams, dataset_mock, config):
124124
def make_msg_and_close(head):
125+
unclosed_stream_reprs = []
125126
unclosed_streams = []
126127
for stream in StreamWrapper.session_streams.keys():
127-
unclosed_streams.append(repr(stream.file_obj))
128+
unclosed_stream_reprs.append(repr(stream.file_obj))
129+
unclosed_streams.append(stream)
130+
for stream in unclosed_streams:
128131
stream.close()
129-
unclosed_streams = "\n".join(unclosed_streams)
130-
return f"{head}\n\n{unclosed_streams}"
132+
unclosed_stream_reprs = "\n".join(unclosed_stream_reprs)
133+
return f"{head}\n\n{unclosed_stream_reprs}"
131134

132135
if StreamWrapper.session_streams:
133136
raise pytest.UsageError(make_msg_and_close("A previous test did not close the following streams:"))

0 commit comments

Comments
 (0)