fix Windows tests

2 files changed, 3 insertions, 2 deletions

diff --git a/tests/unit/test_api.py b/tests/unit/test_api.py
index fc29c33f..4ee19bc4 100644
--- a/tests/unit/test_api.py
+++ b/tests/unit/test_api.py
@@ -87,4 +87,4 @@ def test_sort_code_string_mixed_newlines():
 def test_get_import_file(imperfect, capsys):
     api.get_imports_file(imperfect, sys.stdout)
     out, _ = capsys.readouterr()
-    assert out == imperfect_content
+    assert out == imperfect_content.replace("\n", os.linesep)
diff --git a/tests/unit/test_main.py b/tests/unit/test_main.py
index 93ed94e8..70eafb3a 100644
--- a/tests/unit/test_main.py
+++ b/tests/unit/test_main.py
@@ -1,4 +1,5 @@
 import json
+import os
 import subprocess
 from datetime import datetime
 from io import BytesIO, TextIOWrapper
@@ -925,4 +926,4 @@ def test_identify_imports_main(tmpdir, capsys):
     main.identify_imports_main([str(some_file)])
 
     out, error = capsys.readouterr()
-    assert out == file_imports
+    assert out == file_imports.replace("\n", os.linesep)