From 47364faf5563df0eaa631ed10383817762c6b547 Mon Sep 17 00:00:00 2001 From: Ben Beasley Date: Thu, 5 Jan 2023 18:49:54 -0500 Subject: [PATCH] Work with setuptools_scm 7.1 (fix #25) (#26) 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') -- 2.39.1