100% test coverage

Author: timothycrosley

tests/unit/test_api.py

@@ -5,7 +5,7 @@
 
 import pytest
 
-from isort import api
+from isort import ImportKey, api
 from isort.settings import Config
 
 imperfect_content = "import b\nimport a\n"
@@ -86,3 +86,18 @@ def test_sort_code_string_mixed_newlines():
 def test_find_imports_in_file(imperfect):
     found_imports = list(api.find_imports_in_file(imperfect))
     assert "b" in [found_import.module for found_import in found_imports]
+
+
+def test_find_imports_in_code():
+    code = """
+from x.y import z as a
+from x.y import z as a
+from x.y import z
+import x.y
+import x
+"""
+    assert len(list(api.find_imports_in_code(code))) == 5
+    assert len(list(api.find_imports_in_code(code, unique=True))) == 4
+    assert len(list(api.find_imports_in_code(code, unique=ImportKey.ATTRIBUTE))) == 3
+    assert len(list(api.find_imports_in_code(code, unique=ImportKey.MODULE))) == 2
+    assert len(list(api.find_imports_in_code(code, unique=ImportKey.PACKAGE))) == 1

tests/unit/test_main.py

@@ -1022,3 +1022,15 @@ def test_identify_imports_main(tmpdir, capsys):
     assert out.replace("\r\n", "\n") == file_imports_with_dupes.replace(str(some_file), "")
 
     main.identify_imports_main([str(tmpdir)])
+
+    main.identify_imports_main(["-", "--packages"], stdin=as_stream(file_content))
+    out, error = capsys.readouterr()
+    len(out.split("\n")) == 2
+
+    main.identify_imports_main(["-", "--modules"], stdin=as_stream(file_content))
+    out, error = capsys.readouterr()
+    len(out.split("\n")) == 2
+
+    main.identify_imports_main(["-", "--attributes"], stdin=as_stream(file_content))
+    out, error = capsys.readouterr()
+    len(out.split("\n")) == 2