def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ArrayExtremaTask(task_name='Test')
self.root.children_task.append(self.task)
array = np.zeros((5, ), dtype=[('var1', 'f8'), ('var2', 'f8')])
array['var1'][1] = -1
array['var1'][3] = 1
self.root.write_in_database('array', array)
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ArrayExtremaTask(task_name='Test')
self.root.children_task.append(self.task)
array = np.zeros((5,), dtype=[('var1', 'f8'), ('var2', 'f8')])
array['var1'][1] = -1
array['var1'][3] = 1
self.root.write_in_database('array', array)
def setup(self):
self.workbench = Workbench()
self.workbench.register(CoreManifest())
self.workbench.register(StateManifest())
self.workbench.register(PreferencesManifest())
self.workbench.register(TaskManagerManifest())
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ArrayExtremaTask(task_name='Test')
self.root.children_task.append(self.task)
class TestArrayExtremaTask(object):
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ArrayExtremaTask(task_name='Test')
self.root.children_task.append(self.task)
array = np.zeros((5,), dtype=[('var1', 'f8'), ('var2', 'f8')])
array['var1'][1] = -1
array['var1'][3] = 1
self.root.write_in_database('array', array)
def test_mode_observation(self):
# Check database is correctly updated when the mode change.
self.task.mode = 'Min'
assert_equal(self.task.get_from_database('Test_min_ind'), 0)
assert_equal(self.task.get_from_database('Test_min_value'), 1.0)
aux = self.task.accessible_database_entries()
assert_not_in('Test_max_ind', aux)
assert_not_in('Test_max_value', aux)
self.task.mode = 'Max'
assert_equal(self.task.get_from_database('Test_max_ind'), 0)
assert_equal(self.task.get_from_database('Test_max_value'), 2.0)
aux = self.task.accessible_database_entries()
assert_not_in('Test_min_ind', aux)
assert_not_in('Test_min_value', aux)
self.task.mode = 'Max & min'
assert_equal(self.task.get_from_database('Test_min_ind'), 0)
assert_equal(self.task.get_from_database('Test_min_value'), 1.0)
assert_equal(self.task.get_from_database('Test_max_ind'), 0)
assert_equal(self.task.get_from_database('Test_max_value'), 2.0)
def test_check1(self):
# Simply test that everything is ok if the array exists in the database
self.root.write_in_database('array', np.zeros((5,)))
self.task.target_array = '{Root_array}'
test, traceback = self.task.check()
assert_true(test)
assert_false(traceback)
def test_check2(self):
# Simply test that everything is ok if the array exists in the database
# and the column name is ok.
self.task.target_array = '{Root_array}'
self.task.column_name = 'var1'
test, traceback = self.task.check()
assert_true(test)
assert_false(traceback)
def test_check3(self):
# Test handling a wrong array name.
self.task.target_array = '*{Root_array}'
self.task.column_name = 'var3'
test, traceback = self.task.check()
assert_false(test)
assert_equal(len(traceback), 1)
assert_in('root/Test', traceback)
def test_check4(self):
# Test handling an array without names when a name is given.
self.root.write_in_database('array', np.zeros((5,)))
self.task.target_array = '{Root_array}'
self.task.column_name = 'var1'
test, traceback = self.task.check()
assert_false(test)
assert_equal(len(traceback), 1)
assert_in('root/Test', traceback)
def test_check5(self):
# Test handling an array with names when no name is given.
self.task.target_array = '{Root_array}'
test, traceback = self.task.check()
assert_false(test)
assert_equal(len(traceback), 1)
assert_in('root/Test', traceback)
def test_check6(self):
# Test handling a wrong column name.
self.task.target_array = '{Root_array}'
self.task.column_name = 'var3'
test, traceback = self.task.check()
assert_true(test)
assert_equal(len(traceback), 1)
assert_in('root/Test', traceback)
def test_check7(self):
# Test handling a 2d array without names.
self.task.target_array = '{Root_array}'
array = np.zeros((5, 5))
self.root.write_in_database('array', array)
test, traceback = self.task.check()
assert_false(test)
assert_equal(len(traceback), 1)
assert_in('root/Test', traceback)
def test_perform1(self):
# Test performing when mode is 'Max'.
self.task.mode = 'Max'
self.task.target_array = '{Root_array}'
self.task.column_name = 'var1'
self.root.task_database.prepare_for_running()
self.task.perform()
assert_equal(self.task.get_from_database('Test_max_ind'), 3)
assert_equal(self.task.get_from_database('Test_max_value'), 1.0)
def test_perform2(self):
# Test performing when mode is 'Min'.
self.task.mode = 'Min'
self.task.target_array = '{Root_array}'
self.task.column_name = 'var1'
self.root.task_database.prepare_for_running()
self.task.perform()
assert_equal(self.task.get_from_database('Test_min_ind'), 1)
assert_equal(self.task.get_from_database('Test_min_value'), -1.0)
def test_perform3(self):
# Test performing when mode is 'Max & min'.
self.task.mode = 'Max & min'
self.task.target_array = '{Root_array}'
self.task.column_name = 'var1'
self.root.task_database.prepare_for_running()
self.task.perform()
assert_equal(self.task.get_from_database('Test_max_ind'), 3)
assert_equal(self.task.get_from_database('Test_max_value'), 1.0)
assert_equal(self.task.get_from_database('Test_min_ind'), 1)
assert_equal(self.task.get_from_database('Test_min_value'), -1.0)
def test_perform4(self):
# Test performing when no column name is given.
self.root.write_in_database('array', np.zeros((5,)))
self.task.mode = 'Max'
self.task.target_array = '{Root_array}'
self.root.task_database.prepare_for_running()
self.task.perform()
assert_equal(self.task.get_from_database('Test_max_ind'), 0)
assert_equal(self.task.get_from_database('Test_max_value'), 0.0)
class TestArrayExtremaTask(object):
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ArrayExtremaTask(task_name='Test')
self.root.children_task.append(self.task)
array = np.zeros((5, ), dtype=[('var1', 'f8'), ('var2', 'f8')])
array['var1'][1] = -1
array['var1'][3] = 1
self.root.write_in_database('array', array)
def test_mode_observation(self):
# Check database is correctly updated when the mode change.
self.task.mode = 'Min'
assert_equal(self.task.get_from_database('Test_min_ind'), 0)
assert_equal(self.task.get_from_database('Test_min_value'), 1.0)
aux = self.task.accessible_database_entries()
assert_not_in('Test_max_ind', aux)
assert_not_in('Test_max_value', aux)
self.task.mode = 'Max'
assert_equal(self.task.get_from_database('Test_max_ind'), 0)
assert_equal(self.task.get_from_database('Test_max_value'), 2.0)
aux = self.task.accessible_database_entries()
assert_not_in('Test_min_ind', aux)
assert_not_in('Test_min_value', aux)
self.task.mode = 'Max & min'
assert_equal(self.task.get_from_database('Test_min_ind'), 0)
assert_equal(self.task.get_from_database('Test_min_value'), 1.0)
assert_equal(self.task.get_from_database('Test_max_ind'), 0)
assert_equal(self.task.get_from_database('Test_max_value'), 2.0)
def test_check1(self):
# Simply test that everything is ok if the array exists in the database
self.root.write_in_database('array', np.zeros((5, )))
self.task.target_array = '{Root_array}'
test, traceback = self.task.check()
assert_true(test)
assert_false(traceback)
def test_check2(self):
# Simply test that everything is ok if the array exists in the database
# and the column name is ok.
self.task.target_array = '{Root_array}'
self.task.column_name = 'var1'
test, traceback = self.task.check()
assert_true(test)
assert_false(traceback)
def test_check3(self):
# Test handling a wrong array name.
self.task.target_array = '*{Root_array}'
self.task.column_name = 'var3'
test, traceback = self.task.check()
assert_false(test)
assert_equal(len(traceback), 1)
assert_in('root/Test', traceback)
def test_check4(self):
# Test handling an array without names when a name is given.
self.root.write_in_database('array', np.zeros((5, )))
self.task.target_array = '{Root_array}'
self.task.column_name = 'var1'
test, traceback = self.task.check()
assert_false(test)
assert_equal(len(traceback), 1)
assert_in('root/Test', traceback)
def test_check5(self):
# Test handling an array with names when no name is given.
self.task.target_array = '{Root_array}'
test, traceback = self.task.check()
assert_false(test)
assert_equal(len(traceback), 1)
assert_in('root/Test', traceback)
def test_check6(self):
# Test handling a wrong column name.
self.task.target_array = '{Root_array}'
self.task.column_name = 'var3'
test, traceback = self.task.check()
assert_true(test)
assert_equal(len(traceback), 1)
assert_in('root/Test', traceback)
def test_check7(self):
# Test handling a 2d array without names.
self.task.target_array = '{Root_array}'
array = np.zeros((5, 5))
self.root.write_in_database('array', array)
test, traceback = self.task.check()
assert_false(test)
assert_equal(len(traceback), 1)
assert_in('root/Test', traceback)
def test_perform1(self):
# Test performing when mode is 'Max'.
self.task.mode = 'Max'
self.task.target_array = '{Root_array}'
self.task.column_name = 'var1'
self.root.task_database.prepare_for_running()
self.task.perform()
assert_equal(self.task.get_from_database('Test_max_ind'), 3)
assert_equal(self.task.get_from_database('Test_max_value'), 1.0)
def test_perform2(self):
# Test performing when mode is 'Min'.
self.task.mode = 'Min'
self.task.target_array = '{Root_array}'
self.task.column_name = 'var1'
self.root.task_database.prepare_for_running()
self.task.perform()
assert_equal(self.task.get_from_database('Test_min_ind'), 1)
assert_equal(self.task.get_from_database('Test_min_value'), -1.0)
def test_perform3(self):
# Test performing when mode is 'Max & min'.
self.task.mode = 'Max & min'
self.task.target_array = '{Root_array}'
self.task.column_name = 'var1'
self.root.task_database.prepare_for_running()
self.task.perform()
assert_equal(self.task.get_from_database('Test_max_ind'), 3)
assert_equal(self.task.get_from_database('Test_max_value'), 1.0)
assert_equal(self.task.get_from_database('Test_min_ind'), 1)
assert_equal(self.task.get_from_database('Test_min_value'), -1.0)
def test_perform4(self):
# Test performing when no column name is given.
self.root.write_in_database('array', np.zeros((5, )))
self.task.mode = 'Max'
self.task.target_array = '{Root_array}'
self.root.task_database.prepare_for_running()
self.task.perform()
assert_equal(self.task.get_from_database('Test_max_ind'), 0)
assert_equal(self.task.get_from_database('Test_max_value'), 0.0)
