class TestPNASetRFFrequencyTask(object): def setup(self): self.root = RootTask(should_stop=Event(), should_pause=Event()) self.task = SetRFFrequencyTask(task_name='Test') self.root.children_task.append(self.task) self.root.run_time['drivers'] = {'Test': InstrHelper} self.task.unit = 'GHz' # This is set simply to make sure the test of InstrTask pass. self.task.selected_driver = 'Test' self.task.selected_profile = 'Test1' def test_check_pna_interface1(self): # Simply test that everything is ok if frequency can be evaluated. self.task.interface = PNASetRFFrequencyInterface(task=self.task, channel=1) self.task.frequency = '1.0' profile = {'Test1': ({'defined_channels': [[1]]}, {})} self.root.run_time['profiles'] = profile test, traceback = self.task.check(test_instr=True) assert_true(test) assert_false(traceback) def test_check_pna_interface2(self): # Check handling a wrong channel. self.task.interface = PNASetRFFrequencyInterface(task=self.task, channel=1) self.task.frequency = '1.0' profile = {'Test1': ({'defined_channels': [[2]]}, {})} self.root.run_time['profiles'] = profile test, traceback = self.task.check(test_instr=True) assert_false(test) assert_equal(len(traceback), 1) def test_perform_pna_interface(self): self.task.interface = PNASetRFFrequencyInterface(task=self.task) self.task.frequency = '1.0' profile = { 'Test1': ({ 'frequency': [0.0], 'owner': [None] }, { 'get_channel': lambda x, i: x }) } self.root.run_time['profiles'] = profile self.root.task_database.prepare_for_running() self.task.perform() assert_equal(self.root.get_from_database('Test_frequency'), 1.0e9)
class TestPNASetRFFrequencyTask(object): def setup(self): self.root = RootTask(should_stop=Event(), should_pause=Event()) self.task = SetRFFrequencyTask(task_name='Test') self.root.children_task.append(self.task) self.root.run_time['drivers'] = {'Test': InstrHelper} self.task.unit = 'GHz' # This is set simply to make sure the test of InstrTask pass. self.task.selected_driver = 'Test' self.task.selected_profile = 'Test1' def test_check_pna_interface1(self): # Simply test that everything is ok if frequency can be evaluated. self.task.interface = PNASetRFFrequencyInterface(task=self.task, channel=1) self.task.frequency = '1.0' profile = {'Test1': ({'defined_channels': [[1]]}, {})} self.root.run_time['profiles'] = profile test, traceback = self.task.check(test_instr=True) assert_true(test) assert_false(traceback) def test_check_pna_interface2(self): # Check handling a wrong channel. self.task.interface = PNASetRFFrequencyInterface(task=self.task, channel=1) self.task.frequency = '1.0' profile = {'Test1': ({'defined_channels': [[2]]}, {})} self.root.run_time['profiles'] = profile test, traceback = self.task.check(test_instr=True) assert_false(test) assert_equal(len(traceback), 1) def test_perform_pna_interface(self): self.task.interface = PNASetRFFrequencyInterface(task=self.task) self.task.frequency = '1.0' profile = {'Test1': ({'frequency': [0.0], 'owner': [None]}, {'get_channel': lambda x, i: x} )} self.root.run_time['profiles'] = profile self.root.task_database.prepare_for_running() self.task.perform() assert_equal(self.root.get_from_database('Test_frequency'), 1.0e9)