From f52797bddd6c24e5dffdfd32bef9aa7612975c82 Mon Sep 17 00:00:00 2001 From: Ilya Chistyakov Date: Sat, 8 Jan 2022 13:36:38 +0300 Subject: [PATCH] fix verbose mod --- pytest_neo.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/pytest_neo.py b/pytest_neo.py index 6b56ed0..edaa6dd 100644 --- a/pytest_neo.py +++ b/pytest_neo.py @@ -12,17 +12,17 @@ import collections import curses import itertools +import multiprocessing import os import queue import random import sys -import threading import time import pytest from _pytest.terminal import TerminalReporter -__version__ = '0.2.2' +__version__ = '0.2.3' BLOB_SIZE = (10, 20) BLOB_SPEED = (0.1, 0.2) @@ -258,8 +258,8 @@ def write_fspath_result(self, nodeid, res): self.write_new_column() @pytest.hookimpl(trylast=True) - def pytest_sessionstart(self, session): - super(NeoTerminalReporter, self).pytest_sessionstart(session) + def pytest_collection_finish(self, session): + super(NeoTerminalReporter, self).pytest_collection_finish(session) self.tearup() def pytest_internalerror(self, excrepr): @@ -358,7 +358,7 @@ def can_draw(self, current_time): return current_time - self._last_draw > self.speed -class VerboseReporter(threading.Thread): +class VerboseReporter(multiprocessing.Process): REFRESH_INTERVAL = 0.01 def __init__(self, stdscr, speed_min, speed_max): @@ -369,8 +369,8 @@ def __init__(self, stdscr, speed_min, speed_max): self.speed_min = speed_min self.speed_max = speed_max self._killed = False - self.queue = queue.Queue() - self.exit = threading.Event() + self.queue = multiprocessing.Queue() + self.exit = multiprocessing.Event() def run(self): try: