Make version test use regex (#242)

As per adamchainz/flake8-no-types#28
adamchainz · May 23, 2020 · acb34ce · acb34ce
1 parent bb3902c
commit acb34ce
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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