class TestSetDCVoltageTask(object): def setup(self): self.root = RootTask(should_stop=Event(), should_pause=Event()) self.task = MeasDCVoltageTask(task_name='Test') self.root.children_task.append(self.task) self.root.run_time['drivers'] = {'Test': InstrHelper} # This is set simply to make sure the test of InstrTask pass. self.task.selected_driver = 'Test' self.task.selected_profile = 'Test1' def test_perform(self): self.task.wait_time = 1.0 self.root.run_time['profiles'] = {'Test1': ({}, {'read_voltage_dc': [2.0]}) } self.root.task_database.prepare_for_running() self.task.perform() assert_equal(self.root.get_from_database('Test_voltage'), 2.0)
class TestSetDCVoltageTask(object): def setup(self): self.root = RootTask(should_stop=Event(), should_pause=Event()) self.task = MeasDCVoltageTask(task_name='Test') self.root.children_task.append(self.task) self.root.run_time['drivers'] = {'Test': InstrHelper} # This is set simply to make sure the test of InstrTask pass. self.task.selected_driver = 'Test' self.task.selected_profile = 'Test1' def test_perform(self): self.task.wait_time = 1.0 self.root.run_time['profiles'] = { 'Test1': ({}, { 'read_voltage_dc': [2.0] }) } self.root.task_database.prepare_for_running() self.task.perform() assert_equal(self.root.get_from_database('Test_voltage'), 2.0)