Skip to content

Commit e27a32f

Browse files
committed
fix verify environment to work with prereleases
1 parent bfe832c commit e27a32f

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

tests/verify_environment.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import os
22
import sys
3-
from django import VERSION
3+
import django
44

55

66
def test():
@@ -16,8 +16,11 @@ def test():
1616
f"{expected_python}"
1717
)
1818

19-
expected_django = tuple(int(v) for v in expected_django.split(".") if v)
20-
assert VERSION[: len(expected_django)] == expected_django, (
21-
f"Django Version Mismatch: {VERSION[: len(expected_django)]} != "
22-
f"{expected_django}"
23-
)
19+
try:
20+
expected_django = tuple(int(v) for v in expected_django.split(".") if v)
21+
assert django.VERSION[: len(expected_django)] == expected_django, (
22+
f"Django Version Mismatch: {django.VERSION[: len(expected_django)]} != "
23+
f"{expected_django}"
24+
)
25+
except ValueError:
26+
assert expected_django == django.__version__

0 commit comments

Comments
 (0)