|
1 | 1 | # -*- coding: utf-8 -*-
|
2 | 2 | """Tests for _check_soft_dependencies utility."""
|
| 3 | +from unittest.mock import patch |
| 4 | + |
3 | 5 | import pytest
|
4 | 6 | from packaging.requirements import InvalidRequirement
|
5 | 7 |
|
6 |
| -from skbase.utils.dependencies._dependencies import _check_soft_dependencies |
| 8 | +from skbase.utils.dependencies import _check_python_version, _check_soft_dependencies |
7 | 9 |
|
8 | 10 |
|
9 | 11 | def test_check_soft_deps():
|
@@ -47,3 +49,53 @@ def test_check_soft_deps():
|
47 | 49 | assert _check_soft_dependencies(
|
48 | 50 | ("pytest", "!!numpy<~><>0.1.0"), severity="none"
|
49 | 51 | )
|
| 52 | + |
| 53 | + |
| 54 | +@patch("skbase.utils.dependencies._dependencies.sys") |
| 55 | +@pytest.mark.parametrize( |
| 56 | + "mock_release_version, prereleases, expect_exception", |
| 57 | + [ |
| 58 | + (True, True, False), |
| 59 | + (True, False, True), |
| 60 | + (False, False, False), |
| 61 | + (False, True, False), |
| 62 | + ], |
| 63 | +) |
| 64 | +def test_check_python_version( |
| 65 | + mock_sys, mock_release_version, prereleases, expect_exception |
| 66 | +): |
| 67 | + from skbase.base import BaseObject |
| 68 | + |
| 69 | + if mock_release_version: |
| 70 | + mock_sys.version = "3.8.1rc" |
| 71 | + else: |
| 72 | + mock_sys.version = "3.8.1" |
| 73 | + |
| 74 | + class DummyObjectClass(BaseObject): |
| 75 | + _tags = { |
| 76 | + "python_version": ">=3.7.1", # PEP 440 version specifier, e.g., ">=3.7" |
| 77 | + "python_dependencies": None, # PEP 440 dependency strs, e.g., "pandas>=1.0" |
| 78 | + "env_marker": None, # PEP 508 environment marker, e.g., "os_name=='posix'" |
| 79 | + } |
| 80 | + """Define dummy class to test set_tags.""" |
| 81 | + |
| 82 | + dummy_object_instance = DummyObjectClass() |
| 83 | + |
| 84 | + try: |
| 85 | + _check_python_version(dummy_object_instance, prereleases=prereleases) |
| 86 | + except ModuleNotFoundError as exception: |
| 87 | + expected_msg = ( |
| 88 | + f"{type(dummy_object_instance).__name__} requires python version " |
| 89 | + f"to be {dummy_object_instance.get_tags()['python_version']}, " |
| 90 | + f"but system python version is {mock_sys.version}. " |
| 91 | + "This is due to the release candidate status of your system Python." |
| 92 | + ) |
| 93 | + |
| 94 | + if not expect_exception or exception.msg != expected_msg: |
| 95 | + # Throw Error since exception is not expected or has not the correct message |
| 96 | + raise AssertionError( |
| 97 | + "ModuleNotFoundError should be NOT raised by:", |
| 98 | + f"\n\t - mock_release_version: {mock_release_version},", |
| 99 | + f"\n\t - prereleases: {prereleases},", |
| 100 | + f"\nERROR MESSAGE: {exception.msg}", |
| 101 | + ) from exception |
0 commit comments