From 2bf0d32f92f6609258f85131b94d0952fc0ec149 Mon Sep 17 00:00:00 2001 From: "Benjamin A. Beasley" Date: Mon, 19 Dec 2022 20:20:11 -0500 Subject: [PATCH] Work with setuptools_scm 7.1 (fix #25) Make test_write less brittle (see also #8, #9) so that it works with _version.py files generated by at least setuptools_scm 7.1, 7.0, and 6.x. --- tests/test_build.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/test_build.py b/tests/test_build.py index 2fde601..7c76343 100644 --- a/tests/test_build.py +++ b/tests/test_build.py @@ -75,8 +75,10 @@ def test_write(new_project_write): assert os.path.isfile(version_file) lines = read_file(version_file).splitlines() - assert lines[3].startswith(('version =', '__version__ =')) - assert lines[3].endswith("version = '1.2.3'") + version_starts = ('version = ', '__version__ = ') + assert any(line.startswith(version_starts) for line in lines) + version_line = next(line for line in lines if line.startswith(version_starts)) + assert version_line.endswith(" = '1.2.3'") @pytest.mark.skipif(sys.version_info[0] == 2, reason='Depends on fix in 6.4.0 which is Python 3-only')