From acb34cedddf63d0465715c5bc743b7dee7e29c96 Mon Sep 17 00:00:00 2001
From: Adam Johnson <me@adamj.eu>
Date: Sat, 23 May 2020 19:59:41 +0100
Subject: [PATCH] Make version test use regex (#242)

As per https://github.com/adamchainz/flake8-no-types/pull/28
---
 tests/test_flake8_comprehensions.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_flake8_comprehensions.py b/tests/test_flake8_comprehensions.py
index f35dc39..f80bfab 100644
--- a/tests/test_flake8_comprehensions.py
+++ b/tests/test_flake8_comprehensions.py
@@ -1,3 +1,4 @@
+import re
 import sys
 
 import pytest
@@ -13,9 +14,9 @@
 
 def test_version(flake8dir):
     result = flake8dir.run_flake8(["--version"])
-    version_string = "flake8-comprehensions: " + version("flake8-comprehensions")
+    version_regex = r"flake8-comprehensions:( )*" + version("flake8-comprehensions")
     unwrapped = "".join(result.out_lines)
-    assert version_string in unwrapped
+    assert re.search(version_regex, unwrapped)
 
 
 # C400