コード例 #1
0
ファイル: test_views.py プロジェクト: scottwedge/EXtra-foam
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)
コード例 #2
0
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():