def test_iter_file_tree_chunk4(self):
test_dir_path = os.path.join(TEST_DATA_DIR, 'test_data_file_set_tree')
expected_filepaths = {
os.path.join(test_dir_path, '4/3/2/1/UUID_43210.dataElement'),
}
dfs = DataFileSet(test_dir_path, uuid_chunk=5, pickle_protocol=2)
actual_filepaths = set(dfs._iter_file_tree())
self.assertSetEqual(actual_filepaths, expected_filepaths)
def test_iter_file_tree_chunk4(self):
test_dir_path = os.path.join(TEST_DATA_DIR, 'test_data_file_set_tree')
expected_filepaths = {
os.path.join(test_dir_path, '4/3/2/1/UUID_43210.dataElement'),
}
dfs = DataFileSet(test_dir_path, uuid_chunk=5, pickle_protocol=2)
actual_filepaths = set(dfs._iter_file_tree())
nose.tools.assert_set_equal(actual_filepaths, expected_filepaths)
def test_iter(self, m_open, m_pickle):
expected_file_tree_iter = ['/a', '/b', '/d']
dfs = DataFileSet('/')
dfs._iter_file_tree = mock.MagicMock(
return_value=expected_file_tree_iter)
list(dfs)
nose.tools.assert_equal(m_open.call_count, 3)
nose.tools.assert_equal(m_pickle.load.call_count, 3)
m_open.assert_any_call('/a')
m_open.assert_any_call('/b')
m_open.assert_any_call('/d')
def test_iter_file_tree_chunk0(self):
test_dir_path = os.path.join(TEST_DATA_DIR, 'test_data_file_set_tree')
expected_filepaths = {
os.path.join(test_dir_path, 'UUID_0.dataElement'),
os.path.join(test_dir_path, 'UUID_1.dataElement'),
os.path.join(test_dir_path, 'UUID_2.dataElement'),
}
dfs = DataFileSet(test_dir_path, uuid_chunk=None, pickle_protocol=2)
actual_filepaths = set(dfs._iter_file_tree())
self.assertSetEqual(actual_filepaths, expected_filepaths)
def test_iter(self, m_open, m_pickle):
expected_file_tree_iter = ['/a', '/b', '/d']
dfs = DataFileSet('/')
dfs._iter_file_tree = mock.MagicMock(
return_value=expected_file_tree_iter)
list(dfs)
self.assertEqual(m_open.call_count, 3)
self.assertEqual(m_pickle.load.call_count, 3)
m_open.assert_any_call('/a')
m_open.assert_any_call('/b')
m_open.assert_any_call('/d')
def test_iter_file_tree_chunk3(self):
test_dir_path = os.path.join(TEST_DATA_DIR, 'test_data_file_set_tree')
expected_filepaths = {
os.path.join(test_dir_path, '0/0/UUID_000.dataElement'),
os.path.join(test_dir_path, '0/0/UUID_001.dataElement'),
os.path.join(test_dir_path, '0/1/UUID_012.dataElement'),
os.path.join(test_dir_path, '1/8/UUID_180.dataElement'),
os.path.join(test_dir_path, '3/1/UUID_317.dataElement'),
}
dfs = DataFileSet(test_dir_path, uuid_chunk=3, pickle_protocol=2)
actual_filepaths = set(dfs._iter_file_tree())
nose.tools.assert_set_equal(actual_filepaths, expected_filepaths)
def test_iter_file_tree_chunk3(self):
test_dir_path = os.path.join(TEST_DATA_DIR, 'test_data_file_set_tree')
expected_filepaths = {
os.path.join(test_dir_path, '0/0/UUID_000.dataElement'),
os.path.join(test_dir_path, '0/0/UUID_001.dataElement'),
os.path.join(test_dir_path, '0/1/UUID_012.dataElement'),
os.path.join(test_dir_path, '1/8/UUID_180.dataElement'),
os.path.join(test_dir_path, '3/1/UUID_317.dataElement'),
}
dfs = DataFileSet(test_dir_path, uuid_chunk=3, pickle_protocol=2)
actual_filepaths = set(dfs._iter_file_tree())
self.assertSetEqual(actual_filepaths, expected_filepaths)
def test_count(self):
expected_file_tree_iter = ['/a', '/b', '/d']
dfs = DataFileSet('/')
dfs._iter_file_tree = mock.MagicMock(
return_value=expected_file_tree_iter)
nose.tools.assert_equal(dfs.count(), 3)
def test_count(self):
expected_file_tree_iter = ['/a', '/b', '/d']
dfs = DataFileSet('/')
dfs._iter_file_tree = mock.MagicMock(
return_value=expected_file_tree_iter)
self.assertEqual(dfs.count(), 3)
