diff options
-rw-r--r-- | tests/file_test.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/file_test.py b/tests/file_test.py index 321533b..962eb19 100644 --- a/tests/file_test.py +++ b/tests/file_test.py @@ -22,6 +22,7 @@ from glob import glob from nose.tools import * from os import path, remove +from combox.config import get_nodedirs from combox.file import (split_data, glue_data, write_file, read_file, write_shards, read_shards) @@ -66,7 +67,7 @@ and check if they're the same as the orginal file. f_shards = split_data(f_content, SHARDS) f_basename = path.basename(f) - nodes = [path.abspath(node['path']) for node in config['nodes_info'].itervalues()] + nodes = get_nodedirs(config) write_shards(f_shards, nodes, f_basename) # check if the shards have been created. |