Fix tests

ionelmc · Dec 15, 2021 · 6a73358 · 6a73358
1 parent ba1cba1
commit 6a73358
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/tests/test_tox_wheel.py b/tests/test_tox_wheel.py
@@ -1,3 +1,5 @@
+import sys
+
 import pytest
 
 import tox_wheel.plugin
@@ -83,7 +85,11 @@ def test_enabled_pep517(testdir_pep517, options):
     result.stdout.fnmatch_lines([
         'py* wheel-make: *',
     ])
-    assert result.stdout.str().count('Building wheel for foobar (PEP 517)') == 4
+    if sys.version_info >= (3, 6):
+        build_string = 'Building wheel for foobar (pyproject.toml)'
+    else:
+        build_string = 'Building wheel for foobar (PEP 517)'
+    assert result.stdout.str().count(build_string) == 4
     assert result.ret == 0
 
 
@@ -121,7 +127,11 @@ def test_build_env_pep517(testdir_pep517, options):
     result.stdout.fnmatch_lines([
         'build wheel-make: *',
     ])
-    assert result.stdout.str().count('Building wheel for foobar (PEP 517)') == 2
+    if sys.version_info >= (3, 6):
+        build_string = 'Building wheel for foobar (pyproject.toml)'
+    else:
+        build_string = 'Building wheel for foobar (PEP 517)'
+    assert result.stdout.str().count(build_string) == 2
     assert result.ret == 0