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))
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
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')
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