diff --git a/file_dir_set_ops/main.py b/file_dir_set_ops/main.py index 43bcdfb..59bdebc 100644 --- a/file_dir_set_ops/main.py +++ b/file_dir_set_ops/main.py @@ -10,7 +10,7 @@ def main(): operation_choices = intersection_choices + union_choices + complement_choices arg_parser = argparse.ArgumentParser() - arg_parser.add_argument('--version', '-version', '--v', '-v', action='version', version='0.0.0.3') + arg_parser.add_argument('--version', '-version', '--v', '-v', action='version', version='0.1.0') arg_parser.add_argument('--operation', '-operation', '--op', '-op', action='store', type=str, required=True, choices=operation_choices, help='Set operation to perform.') arg_parser.add_argument('--dir-paths', '-dir-paths', '--paths', '-paths', action='store', type=str, required=True, nargs='+', help='Directory paths for sets of files that will be used in Set operations.') args = arg_parser.parse_args() diff --git a/setup.py b/setup.py index a8f7f58..5392a9d 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ description='Set Operations on File Directories.', long_description='Set Operations on File Directories.', url='https://github.com/njgibbon/file-dir-set-ops', - version='0.0.0.3', + version='0.1.0', author="Nick Gibbon", packages=find_packages(), entry_points={ diff --git a/tests/test_main.py b/tests/test_main.py index 617c499..44d01aa 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -7,3 +7,9 @@ class TestMain(unittest.TestCase): def test_print_items_with_newline(self): items = ['a', 'b'] file_dir_set_ops.main.print_items_with_newline(items) + + def test_get_file_set_from_dir_path(self): + expected_set = {'a.txt', 'file.txt'} + file_set = file_dir_set_ops.main.get_file_set_from_dir_path("tests/data/a") + print(file_set) + assert(file_set == expected_set)