From e6ec382eed4432ef2ab43bd5964c83dc45ac615d Mon Sep 17 00:00:00 2001 From: anirudnits Date: Sat, 10 Oct 2020 20:30:40 +0530 Subject: Added tests for combine_straight_imports option --- tests/unit/test_main.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'tests/unit/test_main.py') diff --git a/tests/unit/test_main.py b/tests/unit/test_main.py index 59e224ae..0fa4cd76 100644 --- a/tests/unit/test_main.py +++ b/tests/unit/test_main.py @@ -77,6 +77,8 @@ def test_parse_args(): assert main.parse_args(["--os"]) == {"only_sections": True} assert main.parse_args(["--om"]) == {"only_modified": True} assert main.parse_args(["--only-modified"]) == {"only_modified": True} + assert main.parse_args(["--csi"]) == {"combine_straight_imports": True} + assert main.parse_args(["--combine-straight-imports"]) == {"combine_straight_imports": True} def test_ascii_art(capsys): @@ -762,6 +764,25 @@ import b assert "else-type place_module for a returned THIRDPARTY" not in out assert "else-type place_module for b returned THIRDPARTY" not in out + # ensures that combine-straight-imports flag works with stdin + input_content = UnseekableTextIOWrapper( + BytesIO( + b""" +import a +import b +""" + ) + ) + + main.main(["-", "--combine-straight-imports"], stdin=input_content) + out, error = capsys.readouterr() + + assert out == ( + """ +import a b +""" + ) + def test_unsupported_encodings(tmpdir, capsys): tmp_file = tmpdir.join("file.py") -- cgit v1.2.1