diff options
| author | Timur Kushukov <timqsh@gmail.com> | 2020-10-16 21:44:27 +0500 |
|---|---|---|
| committer | Timur Kushukov <timqsh@gmail.com> | 2020-10-17 14:16:19 +0500 |
| commit | 579d93c40c736d87f5219c42222df8f02abf3abf (patch) | |
| tree | 84a84aa5e2c4e2cf23b3ae80f19c82585c57a18f /tests/unit/test_main.py | |
| parent | 6807cfbaeb7d25d2b92c4e0efce41371ffff9777 (diff) | |
| download | isort-579d93c40c736d87f5219c42222df8f02abf3abf.tar.gz | |
CLI to get imports
Diffstat (limited to 'tests/unit/test_main.py')
| -rw-r--r-- | tests/unit/test_main.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/unit/test_main.py b/tests/unit/test_main.py index b2035f2c..93ed94e8 100644 --- a/tests/unit/test_main.py +++ b/tests/unit/test_main.py @@ -914,3 +914,15 @@ import os assert "else-type place_module for os returned STDLIB" in out assert "else-type place_module for math returned STDLIB" not in out assert "else-type place_module for pandas returned THIRDPARTY" not in out + + +def test_identify_imports_main(tmpdir, capsys): + file_content = "import mod2\n" "a = 1\n" "import mod1\n" + file_imports = "import mod2\n" "import mod1\n" + some_file = tmpdir.join("some_file.py") + some_file.write(file_content) + + main.identify_imports_main([str(some_file)]) + + out, error = capsys.readouterr() + assert out == file_imports |