diff --git a/minimum_dependency_generator/minimum_dependency_generator.py b/minimum_dependency_generator/minimum_dependency_generator.py index 5d3f899..c98c9f2 100644 --- a/minimum_dependency_generator/minimum_dependency_generator.py +++ b/minimum_dependency_generator/minimum_dependency_generator.py @@ -150,6 +150,7 @@ def generate_min_requirements(paths, options=None, extras_require=None, output_f write_file(min_requirements, output_filepath) return min_requirements + def write_file(data, filepath): try: with open(filepath, 'w') as f: diff --git a/minimum_dependency_generator/tests/test_utils.py b/minimum_dependency_generator/tests/test_utils.py index de5e8e5..c5af522 100644 --- a/minimum_dependency_generator/tests/test_utils.py +++ b/minimum_dependency_generator/tests/test_utils.py @@ -1,6 +1,8 @@ -from ..minimum_dependency_generator import write_file from tempfile import NamedTemporaryFile +from ..minimum_dependency_generator import write_file + + def test_write_text_file(): data = 'scipy==0.8.0\npandas==1.2.0\n' with NamedTemporaryFile() as temp: @@ -10,4 +12,3 @@ def test_write_text_file(): assert len(min_reqs) == 2 assert 'scipy==0.8.0' == min_reqs[0].strip() assert 'pandas==1.2.0' == min_reqs[1].strip() -