From bcbfef8c2eb7bb8df05900128f08beb51955efd3 Mon Sep 17 00:00:00 2001 From: Adam Johnson Date: Sat, 23 May 2020 19:44:33 +0100 Subject: [PATCH 1/2] Improve version test In case of wrapping as in https://github.com/adamchainz/flake8-tidy-imports/pull/205 --- tests/test_flake8_comprehensions.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_flake8_comprehensions.py b/tests/test_flake8_comprehensions.py index 146c9fe..59116fd 100644 --- a/tests/test_flake8_comprehensions.py +++ b/tests/test_flake8_comprehensions.py @@ -14,7 +14,8 @@ def test_version(flake8dir): result = flake8dir.run_flake8(["--version"]) version_string = "flake8-comprehensions: " + version("flake8-comprehensions") - assert version_string in result.out_lines[0] + unwrapped = ''.join(result.out_lines) + assert version_string in unwrapped # C400 From 85c07f3743cd5da4f5753e21c98cdfbbe6d9ddb8 Mon Sep 17 00:00:00 2001 From: Adam Johnson Date: Sat, 23 May 2020 19:45:38 +0100 Subject: [PATCH 2/2] black --- tests/test_flake8_comprehensions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_flake8_comprehensions.py b/tests/test_flake8_comprehensions.py index 59116fd..f35dc39 100644 --- a/tests/test_flake8_comprehensions.py +++ b/tests/test_flake8_comprehensions.py @@ -14,7 +14,7 @@ def test_version(flake8dir): result = flake8dir.run_flake8(["--version"]) version_string = "flake8-comprehensions: " + version("flake8-comprehensions") - unwrapped = ''.join(result.out_lines) + unwrapped = "".join(result.out_lines) assert version_string in unwrapped