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(