Example #1
0
def test_pna_power_view(windows, root_view, task_workbench):
    """Test PNA power interface widget outisde of a LoopTask.

    """
    task = SetRFPowerTask(name='Test')
    task.interface = PNASetRFPowerInterface(task=task)
    root_view.task.add_child_task(0, task)
    show_and_close_widget(RFPowerView(task=task, root=root_view))
Example #2
0
class TestSetRFPowerTask(object):

    def setup(self):
        self.root = RootTask(should_stop=Event(), should_pause=Event())
        self.task = SetRFPowerTask(name='Test')
        self.root.add_child_task(0, self.task)

        self.root.run_time[DRIVERS] = {'Test': (InstrHelper,
                                                InstrHelperStarter())}
        self.root.run_time[PROFILES] =\
            {'Test1': {'connections': {'C': {'owner': []}},
                       'settings': {'S': {'check_connection': [True]}}
                       }
             }

        # This is set simply to make sure the test of InstrTask pass.
        self.task.selected_instrument = ('Test1', 'Test', 'C', 'S')

    def test_check_base_interface1(self):
        # Simply test that everything is ok if voltage can be evaluated.
        self.task.power = '1.0'

        test, traceback = self.task.check(test_instr=True)
        assert test
        assert not traceback

    def test_check_base_interface2(self):
        # Check handling a wrong power.
        self.task.power = '*1.0*'

        test, traceback = self.task.check(test_instr=True)
        assert not test
        assert len(traceback) == 1

    def test_perform_base_interface(self):
        self.task.power = '1.0'

        c = self.root.run_time[PROFILES]['Test1']['connections']
        c['C'] = {'power': [0.0], 'owner': [None]}
        self.root.prepare()

        self.task.perform()
        assert self.root.get_from_database('Test_power') == 1.0
Example #3
0
    def setup(self):
        self.root = RootTask(should_stop=Event(), should_pause=Event())
        self.task = SetRFPowerTask(name='Test')
        self.root.add_child_task(0, self.task)

        self.root.run_time[DRIVERS] = {'Test': (InstrHelper,
                                                InstrHelperStarter())}
        self.root.run_time[PROFILES] =\
            {'Test1': {'connections': {'C': {'owner': []}},
                       'settings': {'S': {'check_connection': [True]}}
                       }
             }

        # This is set simply to make sure the test of InstrTask pass.
        self.task.selected_instrument = ('Test1', 'Test', 'C', 'S')
Example #4
0
class TestPNASetRFPowerTask(object):
    """Test of the PNA set power interface.

    """

    def setup(self):
        self.root = RootTask(should_stop=Event(), should_pause=Event())
        self.task = SetRFPowerTask(name='Test')
        self.root.add_child_task(0, self.task)

        self.root.run_time[DRIVERS] = {'Test': (InstrHelper,
                                                InstrHelperStarter())}
        self.root.run_time[PROFILES] =\
            {'Test1': {'connections': {'C': {'owner': []}},
                       'settings': {'S': {'check_connection': [True]}}
                       }
             }

        # This is set simply to make sure the test of InstrTask pass.
        self.task.selected_instrument = ('Test1', 'Test', 'C', 'S')

    def test_check_pna_interface1(self):
        """Simply test that everything is ok if power can be evaluated.

        """
        self.task.interface = PNASetRFPowerInterface(task=self.task,
                                                     channel=1)
        self.task.power = '1.0'
        c = self.root.run_time[PROFILES]['Test1']['connections']
        c['C'] = {'defined_channels': [[1]]}

        test, traceback = self.task.check(test_instr=True)
        assert test
        assert not traceback

    def test_check_pna_interface2(self):
        """Check handling a wrong channel.

        """
        self.task.interface = PNASetRFPowerInterface(task=self.task,
                                                     channel=2)
        self.task.power = '1.0'
        c = self.root.run_time[PROFILES]['Test1']['connections']
        c['C'] = {'defined_channels': [[1]]}

        test, traceback = self.task.check(test_instr=True)
        assert not test
        assert len(traceback) == 1

    def test_perform_pna_interface(self):
        self.task.interface = PNASetRFPowerInterface(task=self.task,
                                                     channel=1)
        self.task.power = '1.0'

        c = self.root.run_time[PROFILES]['Test1']['connections']
        c['C'] = {'power': [0.0], 'port': [1], 'owner': [None]}
        s = self.root.run_time[PROFILES]['Test1']['settings']
        s['S'] = {'get_channel': lambda x, i: x}

        self.root.prepare()

        self.task.perform()
        assert self.root.get_from_database('Test_power') == 1.0