def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = SaveTask(task_name='Test')
self.root.children_task.append(self.task)
self.root.write_in_database('int', 1)
self.root.write_in_database('float', 2.0)
self.root.write_in_database('str', 'a')
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = SaveTask(task_name='Test')
self.root.children_task.append(self.task)
self.root.write_in_database('int', 1)
self.root.write_in_database('float', 2.0)
self.root.write_in_database('str', 'a')
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 = SaveTask(task_name='Test')
self.root.children_task.append(self.task)
class TestSaveTask(object):
test_dir = TEST_PATH
@classmethod
def setup_class(cls):
print complete_line(
__name__ + ':{}.setup_class()'.format(cls.__name__), '-', 77)
os.mkdir(cls.test_dir)
@classmethod
def teardown_class(cls):
print complete_line(
__name__ + ':{}.teardown_class()'.format(cls.__name__), '-', 77)
# Removing pref files creating during tests.
try:
shutil.rmtree(cls.test_dir)
# Hack for win32.
except OSError:
print 'OSError'
dirs = os.listdir(cls.test_dir)
for directory in dirs:
shutil.rmtree(os.path.join(cls.test_dir), directory)
shutil.rmtree(cls.test_dir)
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = SaveTask(task_name='Test')
self.root.children_task.append(self.task)
self.root.write_in_database('int', 1)
self.root.write_in_database('float', 2.0)
self.root.write_in_database('str', 'a')
def teardown(self):
folder = self.test_dir
for the_file in os.listdir(folder):
file_path = os.path.join(folder, the_file)
if os.path.isfile(file_path):
os.remove(file_path)
def test_saving_target_observer(self):
self.task.saving_target = 'Array'
assert_equal(self.task.get_from_database('Test_array'),
np.array([1.0]))
self.task.saving_target = 'File'
aux = self.task.accessible_database_entries()
assert_not_in('Test_array', aux)
self.task.saving_target = 'File and array'
assert_equal(self.task.get_from_database('Test_array'),
np.array([1.0]))
def test_check1(self):
# Test everything ok in file mode (no array size).
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir
task.filename = 'test{Root_int}.txt'
task.file_mode = 'New'
task.header = 'test'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
file_path = os.path.join(self.test_dir, 'test1.txt')
test, traceback = task.check()
assert_true(test)
assert_false(traceback)
assert_false(os.path.isfile(file_path))
assert_false(task.initialized)
task.file_mode = 'Add'
test, traceback = task.check()
assert_true(test)
assert_false(traceback)
assert_true(os.path.isfile(file_path))
os.remove(file_path)
def test_check2(self):
# Test everything of in array mode (assert database state).
task = self.task
task.saving_target = 'Array'
task.array_size = '1000*{Root_float}'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
test, traceback = task.check()
assert_true(test)
assert_false(traceback)
array = task.get_from_database('Test_array')
assert_equal(array.dtype.names, ('toto', 'tata'))
def test_check3(self):
# Test everything is ok in file & array mode.
task = self.task
task.saving_target = 'File and array'
task.folder = self.test_dir
task.filename = 'test_rr.txt'
task.file_mode = 'New'
task.header = 'test'
task.array_size = '1000*{Root_float}'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
file_path = os.path.join(self.test_dir, 'test_rr.txt')
test, traceback = task.check()
assert_true(test)
assert_false(traceback)
assert_false(os.path.isfile(file_path))
array = task.get_from_database('Test_array')
assert_equal(array.dtype.names, ('toto', 'tata'))
def test_check4(self):
# Test check issues in file mode : folder.
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir + '{tt}'
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 1)
def test_check5(self):
# Test check issues in file mode : file.
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir
task.filename = 'test{tt}.txt'
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 1)
def test_check6(self):
# Test check issues in file mode : array_size.
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir + '{tt}'
task.filename = 'test.txt'
task.file_mode = 'New'
task.header = 'test'
task.array_size = '1000*'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
file_path = os.path.join(self.test_dir, 'test.txt')
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 1)
assert_false(os.path.isfile(file_path))
def test_check7(self):
# Test check issues in array mode : wrong array_size.
task = self.task
task.saving_target = 'Array'
task.array_size = '1000*{Root_float}*'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 1)
assert_equal(self.task.get_from_database('Test_array'),
np.array([1.0]))
def test_check8(self):
# Test check issues in array mode : absent array_size.
task = self.task
task.saving_target = 'Array'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 1)
assert_equal(self.task.get_from_database('Test_array'),
np.array([1.0]))
def test_check9(self):
# Test check issues in entrie.
task = self.task
task.saving_target = 'Array'
task.array_size = '1000*{Root_float}'
task.saved_values = [('toto', '{Root_str*}'),
('tata', '{Root_float*}')]
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 2)
array = task.get_from_database('Test_array')
assert_equal(array.dtype.names, ('toto', 'tata'))
def test_check10(self):
# Test warning in case the file already exists in new mode.
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir
task.filename = 'test_e.txt'
task.file_mode = 'New'
task.header = 'test'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
file_path = os.path.join(self.test_dir, 'test_e.txt')
with open(file_path, 'w'):
pass
assert_true(os.path.isfile(file_path))
test, traceback = task.check()
assert_true(test)
assert_true(traceback)
assert_true(os.path.isfile(file_path))
def test_perform1(self):
# Test performing in mode file. (Call three times perform)
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir
task.filename = 'test_perform{Root_int}.txt'
task.file_mode = 'Add'
task.header = 'test'
task.array_size = '3'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
file_path = os.path.join(self.test_dir, 'test_perform1.txt')
with open(file_path, 'w') as f:
f.write('test\n')
task.perform()
assert_true(task.initialized)
assert_true(task.file_object)
assert_equal(task.line_index, 1)
with open(file_path) as f:
a = f.readlines()
assert_equal(a, ['test\n', '# test\n', 'toto\ttata\n', 'a\t2.0\n'])
task.perform()
assert_true(task.initialized)
assert_equal(task.line_index, 2)
with open(file_path) as f:
a = f.readlines()
assert_equal(
a,
['test\n', '# test\n', 'toto\ttata\n', 'a\t2.0\n', 'a\t2.0\n'])
task.perform()
assert_false(task.initialized)
assert_equal(task.line_index, 3)
with open(file_path) as f:
a = f.readlines()
assert_equal(a, [
'test\n', '# test\n', 'toto\ttata\n', 'a\t2.0\n', 'a\t2.0\n',
'a\t2.0\n'
])
def test_perform2(self):
# Test performing in array mode. (Call three times perform)
task = self.task
task.saving_target = 'Array'
task.array_size = '3'
task.saved_values = [('toto', '{Root_int}'), ('tata', '{Root_float}')]
task.perform()
assert_true(task.initialized)
assert_equal(task.line_index, 1)
task.perform()
assert_true(task.initialized)
assert_equal(task.line_index, 2)
task.perform()
assert_false(task.initialized)
assert_equal(task.line_index, 3)
array_type = np.dtype([(str(s[0]), 'f8') for s in task.saved_values])
array = np.empty((3), dtype=array_type)
array[0] = (1, 2.0)
array[1] = (1, 2.0)
array[2] = (1, 2.0)
np.testing.assert_array_equal(task.array, array)
class TestSaveTask(object):
test_dir = TEST_PATH
@classmethod
def setup_class(cls):
print complete_line(__name__ +
':{}.setup_class()'.format(cls.__name__), '-', 77)
os.mkdir(cls.test_dir)
@classmethod
def teardown_class(cls):
print complete_line(__name__ +
':{}.teardown_class()'.format(cls.__name__), '-',
77)
# Removing pref files creating during tests.
try:
shutil.rmtree(cls.test_dir)
# Hack for win32.
except OSError:
print 'OSError'
dirs = os.listdir(cls.test_dir)
for directory in dirs:
shutil.rmtree(os.path.join(cls.test_dir), directory)
shutil.rmtree(cls.test_dir)
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = SaveTask(task_name='Test')
self.root.children_task.append(self.task)
self.root.write_in_database('int', 1)
self.root.write_in_database('float', 2.0)
self.root.write_in_database('str', 'a')
def teardown(self):
folder = self.test_dir
for the_file in os.listdir(folder):
file_path = os.path.join(folder, the_file)
if os.path.isfile(file_path):
os.remove(file_path)
def test_saving_target_observer(self):
self.task.saving_target = 'Array'
assert_equal(self.task.get_from_database('Test_array'),
np.array([1.0]))
self.task.saving_target = 'File'
aux = self.task.accessible_database_entries()
assert_not_in('Test_array', aux)
self.task.saving_target = 'File and array'
assert_equal(self.task.get_from_database('Test_array'),
np.array([1.0]))
def test_check1(self):
# Test everything ok in file mode (no array size).
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir
task.filename = 'test{Root_int}.txt'
task.file_mode = 'New'
task.header = 'test'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
file_path = os.path.join(self.test_dir, 'test1.txt')
test, traceback = task.check()
assert_true(test)
assert_false(traceback)
assert_false(os.path.isfile(file_path))
assert_false(task.initialized)
task.file_mode = 'Add'
test, traceback = task.check()
assert_true(test)
assert_false(traceback)
assert_true(os.path.isfile(file_path))
os.remove(file_path)
def test_check2(self):
# Test everything of in array mode (assert database state).
task = self.task
task.saving_target = 'Array'
task.array_size = '1000*{Root_float}'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
test, traceback = task.check()
assert_true(test)
assert_false(traceback)
array = task.get_from_database('Test_array')
assert_equal(array.dtype.names, ('toto', 'tata'))
def test_check3(self):
# Test everything is ok in file & array mode.
task = self.task
task.saving_target = 'File and array'
task.folder = self.test_dir
task.filename = 'test_rr.txt'
task.file_mode = 'New'
task.header = 'test'
task.array_size = '1000*{Root_float}'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
file_path = os.path.join(self.test_dir, 'test_rr.txt')
test, traceback = task.check()
assert_true(test)
assert_false(traceback)
assert_false(os.path.isfile(file_path))
array = task.get_from_database('Test_array')
assert_equal(array.dtype.names, ('toto', 'tata'))
def test_check4(self):
# Test check issues in file mode : folder.
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir + '{tt}'
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 1)
def test_check5(self):
# Test check issues in file mode : file.
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir
task.filename = 'test{tt}.txt'
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 1)
def test_check6(self):
# Test check issues in file mode : array_size.
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir + '{tt}'
task.filename = 'test.txt'
task.file_mode = 'New'
task.header = 'test'
task.array_size = '1000*'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
file_path = os.path.join(self.test_dir, 'test.txt')
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 1)
assert_false(os.path.isfile(file_path))
def test_check7(self):
# Test check issues in array mode : wrong array_size.
task = self.task
task.saving_target = 'Array'
task.array_size = '1000*{Root_float}*'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 1)
assert_equal(self.task.get_from_database('Test_array'),
np.array([1.0]))
def test_check8(self):
# Test check issues in array mode : absent array_size.
task = self.task
task.saving_target = 'Array'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 1)
assert_equal(self.task.get_from_database('Test_array'),
np.array([1.0]))
def test_check9(self):
# Test check issues in entrie.
task = self.task
task.saving_target = 'Array'
task.array_size = '1000*{Root_float}'
task.saved_values = [('toto', '{Root_str*}'),
('tata', '{Root_float*}')]
test, traceback = task.check()
assert_false(test)
assert_true(traceback)
assert_equal(len(traceback), 2)
array = task.get_from_database('Test_array')
assert_equal(array.dtype.names, ('toto', 'tata'))
def test_check10(self):
# Test warning in case the file already exists in new mode.
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir
task.filename = 'test_e.txt'
task.file_mode = 'New'
task.header = 'test'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
file_path = os.path.join(self.test_dir, 'test_e.txt')
with open(file_path, 'w'):
pass
assert_true(os.path.isfile(file_path))
test, traceback = task.check()
assert_true(test)
assert_true(traceback)
assert_true(os.path.isfile(file_path))
def test_perform1(self):
# Test performing in mode file. (Call three times perform)
task = self.task
task.saving_target = 'File'
task.folder = self.test_dir
task.filename = 'test_perform{Root_int}.txt'
task.file_mode = 'Add'
task.header = 'test'
task.array_size = '3'
task.saved_values = [('toto', '{Root_str}'), ('tata', '{Root_float}')]
file_path = os.path.join(self.test_dir, 'test_perform1.txt')
with open(file_path, 'w') as f:
f.write('test\n')
task.perform()
assert_true(task.initialized)
assert_true(task.file_object)
assert_equal(task.line_index, 1)
with open(file_path) as f:
a = f.readlines()
assert_equal(a, ['test\n', '# test\n', 'toto\ttata\n', 'a\t2.0\n'])
task.perform()
assert_true(task.initialized)
assert_equal(task.line_index, 2)
with open(file_path) as f:
a = f.readlines()
assert_equal(a, ['test\n', '# test\n', 'toto\ttata\n', 'a\t2.0\n',
'a\t2.0\n'])
task.perform()
assert_false(task.initialized)
assert_equal(task.line_index, 3)
with open(file_path) as f:
a = f.readlines()
assert_equal(a, ['test\n', '# test\n', 'toto\ttata\n', 'a\t2.0\n',
'a\t2.0\n', 'a\t2.0\n'])
def test_perform2(self):
# Test performing in array mode. (Call three times perform)
task = self.task
task.saving_target = 'Array'
task.array_size = '3'
task.saved_values = [('toto', '{Root_int}'), ('tata', '{Root_float}')]
task.perform()
assert_true(task.initialized)
assert_equal(task.line_index, 1)
task.perform()
assert_true(task.initialized)
assert_equal(task.line_index, 2)
task.perform()
assert_false(task.initialized)
assert_equal(task.line_index, 3)
array_type = np.dtype([(str(s[0]), 'f8')
for s in task.saved_values])
array = np.empty((3), dtype=array_type)
array[0] = (1, 2.0)
array[1] = (1, 2.0)
array[2] = (1, 2.0)
np.testing.assert_array_equal(task.array, array)
