def main():
parser = argparse.ArgumentParser()
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument(
'-f', '--file', metavar='test_file', nargs=1, help='test file', required=False
)
group.add_argument(
'-d', '--directory', metavar='test_directory', nargs=1, help='directory of test files'
)
args = parser.parse_args()
user_input = args.directory[0] if args.directory else args.file[0]
integration_config.load_integrations(user_input)
threader.launch_threads(
threader.build_threads(
registrar.register(loader.load(user_input))
)
)
from expects import expect
from spec.custom_matchers.contain_exactly_function_called import contain_exactly_function_called
from pysellus import loader
with description('the loader module loads all top-level functions in a directory or file'):
with it('should load every function when there is only one file'):
expect(loader.load('spec/fixtures/one_file/')).to(
contain_exactly_function_called('a_function',
'another_function')
)
with it('should load every function of the specified file'):
expect(loader.load('spec/fixtures/multiple_files/file_1.py')).to(
contain_exactly_function_called('file_1_function_a',
'file_1_function_b')
)
with it('should load every function when there is more than one file'):
expect(loader.load('spec/fixtures/multiple_files/')).to(
contain_exactly_function_called('file_1_function_a',
'file_1_function_b',
'file_2_function_a',
'file_2_function_b',
'file_3_function_a',
'file_3_function_b')
)
with it("should only load setup functions"):
expect(loader.load('spec/fixtures/file_with_helper_functions/')).to(
