better test

alteryx · Mar 21, 2022 · 5a511d7 · 5a511d7
1 parent b0b920a
commit 5a511d7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/minimum_dependency_generator/tests/test_utils.py b/minimum_dependency_generator/tests/test_utils.py
@@ -4,9 +4,10 @@
 
 
 def test_write_text_file():
-    data = 'scipy==0.8.0\npandas==1.2.0\n'
+    reqs = ['scipy==0.8.0', 'pandas==1.2.0']
+    reqs = '\n'.join(reqs) + '\n'
     with NamedTemporaryFile() as temp:
-        write_file(data, temp.name)
+        write_file(reqs, temp.name)
         with open(temp.name) as written:
             min_reqs = written.readlines()
             assert len(min_reqs) == 2