46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
"""Tests for the minimum dependencies in the README.rst file."""
|
|
|
|
|
|
import os
|
|
import re
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
import sklearn
|
|
from sklearn._min_dependencies import dependent_packages
|
|
from sklearn.utils.fixes import parse_version
|
|
|
|
|
|
def test_min_dependencies_readme():
|
|
# Test that the minimum dependencies in the README.rst file are
|
|
# consistent with the minimum dependencies defined at the file:
|
|
# sklearn/_min_dependencies.py
|
|
|
|
pattern = re.compile(r"(\.\. \|)" +
|
|
r"(([A-Za-z]+\-?)+)" +
|
|
r"(MinVersion\| replace::)" +
|
|
r"( [0-9]+\.[0-9]+(\.[0-9]+)?)")
|
|
|
|
readme_path = Path(sklearn.__path__[0]).parents[0]
|
|
readme_file = readme_path / "README.rst"
|
|
|
|
if not os.path.exists(readme_file):
|
|
# Skip the test if the README.rst file is not available.
|
|
# For instance, when installing scikit-learn from wheels
|
|
pytest.skip("The README.rst file is not available.")
|
|
|
|
with readme_file.open("r") as f:
|
|
for line in f:
|
|
matched = pattern.match(line)
|
|
|
|
if not matched:
|
|
continue
|
|
|
|
package, version = matched.group(2), matched.group(5)
|
|
|
|
if package in dependent_packages:
|
|
version = parse_version(version)
|
|
min_version = parse_version(dependent_packages[package][0])
|
|
|
|
assert version == min_version
|