def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = LoadArrayTask(task_name='Test')
self.task.interface = CSVLoadInterface()
self.task.folder = FOLDER_PATH
self.task.filename = 'fake.dat'
self.root.children_task.append(self.task)
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = LoadArrayTask(task_name='Test')
self.task.interface = CSVLoadInterface()
self.task.folder = FOLDER_PATH
self.task.filename = 'fake.dat'
self.root.children_task.append(self.task)
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 = LoadArrayTask(task_name='Test')
self.root.children_task.append(self.task)
class TestLoadArrayTaskCSVInterface(object):
@classmethod
def setup_class(cls):
cls.data = np.zeros((5,), dtype=[('Freq', 'f8'), ('Log', 'f8')])
full_path = os.path.join(FOLDER_PATH, 'fake.dat')
with open(full_path, 'wb') as f:
f.write('# this is a comment \n')
f.write('\t'.join(cls.data.dtype.names) + '\n')
np.savetxt(f, cls.data, delimiter='\t')
@classmethod
def teardown_class(cls):
full_path = os.path.join(FOLDER_PATH, 'fake.dat')
if os.path.isfile(full_path):
os.remove(full_path)
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = LoadArrayTask(task_name='Test')
self.task.interface = CSVLoadInterface()
self.task.folder = FOLDER_PATH
self.task.filename = 'fake.dat'
self.root.children_task.append(self.task)
def test_check1(self):
# Test everything is ok if folder and filename are correct.
test, traceback = self.task.check()
assert_true(test)
assert_false(traceback)
array = self.task.get_from_database('Test_array')
assert_equal(array.dtype.names, ('Freq', 'Log'))
def test_check2(self):
# Test handling wrong folder and filename.
self.task.folder = '{rr}'
self.task.filename = '{tt}'
test, traceback = self.task.check()
assert_false(test)
assert_equal(len(traceback), 2)
def test_check3(self):
# Test handling an absent file.
self.task.filename = 'tt'
test, traceback = self.task.check()
assert_true(test)
assert_equal(len(traceback), 1)
def test_perform1(self):
# Test loading a csv file.
self.task.perform()
array = self.task.get_from_database('Test_array')
np.testing.assert_array_equal(array, self.data)
class TestLoadArrayTaskCSVInterface(object):
@classmethod
def setup_class(cls):
cls.data = np.zeros((5, ), dtype=[('Freq', 'f8'), ('Log', 'f8')])
full_path = os.path.join(FOLDER_PATH, 'fake.dat')
with open(full_path, 'wb') as f:
f.write('# this is a comment \n')
f.write('\t'.join(cls.data.dtype.names) + '\n')
np.savetxt(f, cls.data, delimiter='\t')
@classmethod
def teardown_class(cls):
full_path = os.path.join(FOLDER_PATH, 'fake.dat')
if os.path.isfile(full_path):
os.remove(full_path)
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = LoadArrayTask(task_name='Test')
self.task.interface = CSVLoadInterface()
self.task.folder = FOLDER_PATH
self.task.filename = 'fake.dat'
self.root.children_task.append(self.task)
def test_check1(self):
# Test everything is ok if folder and filename are correct.
test, traceback = self.task.check()
assert_true(test)
assert_false(traceback)
array = self.task.get_from_database('Test_array')
assert_equal(array.dtype.names, ('Freq', 'Log'))
def test_check2(self):
# Test handling wrong folder and filename.
self.task.folder = '{rr}'
self.task.filename = '{tt}'
test, traceback = self.task.check()
assert_false(test)
assert_equal(len(traceback), 2)
def test_check3(self):
# Test handling an absent file.
self.task.filename = 'tt'
test, traceback = self.task.check()
assert_true(test)
assert_equal(len(traceback), 1)
def test_perform1(self):
# Test loading a csv file.
self.task.perform()
array = self.task.get_from_database('Test_array')
np.testing.assert_array_equal(array, self.data)
