import unittest from unittest.mock import MagicMock, patch from PyQt5.QtWidgets import QMainWindow, QWidget from extra_foam.pipeline.tests import _TestDataMixin from extra_foam.pipeline.data_model import ProcessedData from extra_foam.gui import mkQApp from extra_foam.logger import logger app = mkQApp() logger.setLevel('CRITICAL') class TestViews(_TestDataMixin, unittest.TestCase): @classmethod def setUpClass(cls): cls.gui = QMainWindow() # dummy MainGUI cls.gui.createCtrlWidget = MagicMock(return_value=QWidget()) @classmethod def tearDownClass(cls): cls.gui.close() def testCorrectedView(self): from extra_foam.gui.image_tool.corrected_view import CorrectedView # pulse-resolved view = CorrectedView(pulse_resolved=True, parent=self.gui)
import unittest from unittest.mock import patch import time from redis.client import PubSub, Redis from extra_foam.logger import logger from extra_foam.services import start_redis_server from extra_foam.ipc import (init_redis_connection, redis_connection, RedisConnection, RedisSubscriber, RedisPSubscriber, _global_connections) from extra_foam.pipeline.worker import ProcessWorker from extra_foam.processes import wait_until_redis_shutdown logger.setLevel("CRITICAL") class TestRedisConnection(unittest.TestCase): @classmethod def setUpClass(cls): cls._port = 6390 # use a port that is not used in other unittests start_redis_server('127.0.0.1', cls._port) @classmethod def tearDownClass(cls): wait_until_redis_shutdown() def setUp(self): self._db = init_redis_connection('127.0.0.1', self._port) for c in self._db.client_list():