Add additional test cases

PyCQA · Jul 9, 2020 · 2120836 · 2120836
1 parent 2b57c9e
commit 2120836
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/test_isort.py b/tests/test_isort.py
@@ -138,6 +138,19 @@ def test_correct_space_between_imports() -> None:
     test_output_assign = api.sort_code_string(test_input_assign)
     assert test_output_assign == "import sys\n\nVAR = 1\n"
 
+    test_input_assign = "import sys\nVAR = 1\ndef y():\n"
+    test_output_assign = api.sort_code_string(test_input_assign)
+    assert test_output_assign == "import sys\n\nVAR = 1\ndef y():\n"
+
+    test_input = """
+import os
+
+x = "hi"
+def x():
+    pass
+"""
+    assert isort.code(test_input) == test_input
+
 
 def test_sort_on_number() -> None:
     """Ensure numbers get sorted logically (10 > 9 not the other way around)"""