def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ApplyMagFieldTask(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 setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ApplyMagFieldTask(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 setup(self):
self.workbench = Workbench()
self.workbench.register(CoreManifest())
self.workbench.register(StateManifest())
self.workbench.register(PreferencesManifest())
self.workbench.register(InstrManagerManifest())
self.workbench.register(TaskManagerManifest())
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ApplyMagFieldTask(task_name='Test')
self.root.children_task.append(self.task)
self.root.run_time['drivers'] = {'Test': InstrHelper}
class TestApplyMagFieldTask(object):
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ApplyMagFieldTask(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_check1(self):
# Simply test that everything is ok if field can be evaluated.
self.task.target_field = '3.0'
test, traceback = self.task.check(test_instr=True)
assert_true(test)
assert_false(traceback)
assert_equal(self.task.get_from_database('Test_Bfield'), 3.0)
def test_check2(self):
# Check handling a wrong field.
self.task.target_field = '*1.0*'
test, traceback = self.task.check(test_instr=True)
assert_false(test)
assert_equal(len(traceback), 1)
assert_in('root/Test-field', traceback)
assert_equal(self.task.get_from_database('Test_Bfield'), 0.01)
def test_perform1(self):
# Simple test when everything is right.
self.task.target_field = '2.0'
self.root.run_time['profiles'] = {'Test1': ({'owner': []},
{'make_ready': [None],
'go_to_field': [None]}
)}
self.root.task_database.prepare_for_running()
self.task.perform()
join_threads(self.root)
assert_equal(self.root.get_from_database('Test_Bfield'), 2.0)
def test_perform2(self):
# Test multiple run when connection is maintained.
self.task.target_field = '2.0'
self.root.run_time['profiles'] = {'Test1': ({'owner': []},
{'make_ready': [None],
'go_to_field': [None],
'check_connection': [True]
}
)}
self.root.task_database.prepare_for_running()
self.task.perform()
join_threads(self.root)
self.task.perform()
join_threads(self.root)
# In case of fail make_ready would be called twice.
assert_equal(self.root.get_from_database('Test_Bfield'), 2.0)
class TestApplyMagFieldTask(object):
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ApplyMagFieldTask(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_check1(self):
# Simply test that everything is ok if field can be evaluated.
self.task.target_field = '3.0'
test, traceback = self.task.check(test_instr=True)
assert_true(test)
assert_false(traceback)
assert_equal(self.task.get_from_database('Test_Bfield'), 3.0)
def test_check2(self):
# Check handling a wrong field.
self.task.target_field = '*1.0*'
test, traceback = self.task.check(test_instr=True)
assert_false(test)
assert_equal(len(traceback), 1)
assert_in('root/Test-field', traceback)
assert_equal(self.task.get_from_database('Test_Bfield'), 0.01)
def test_perform1(self):
# Simple test when everything is right.
self.task.target_field = '2.0'
self.root.run_time['profiles'] = {
'Test1': ({
'owner': []
}, {
'make_ready': [None],
'go_to_field': [None]
})
}
self.root.task_database.prepare_for_running()
self.task.perform()
join_threads(self.root)
assert_equal(self.root.get_from_database('Test_Bfield'), 2.0)
def test_perform2(self):
# Test multiple run when connection is maintained.
self.task.target_field = '2.0'
self.root.run_time['profiles'] = {
'Test1': ({
'owner': []
}, {
'make_ready': [None],
'go_to_field': [None],
'check_connection': [True]
})
}
self.root.task_database.prepare_for_running()
self.task.perform()
join_threads(self.root)
self.task.perform()
join_threads(self.root)
# In case of fail make_ready would be called twice.
assert_equal(self.root.get_from_database('Test_Bfield'), 2.0)