wehub-resource-sync
98 lines
2.7 KiB
Python
98 lines
2.7 KiB
Python
import sys
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from ray._common.deprecation import (
|
|
DEPRECATED_VALUE,
|
|
Deprecated,
|
|
deprecation_warning,
|
|
)
|
|
|
|
|
|
def test_deprecation_warning_warn():
|
|
with patch("ray._common.deprecation.logger.warning") as mock_warning:
|
|
deprecation_warning("old_feature", "new_feature")
|
|
|
|
mock_warning.assert_called_once()
|
|
args, _ = mock_warning.call_args
|
|
assert (
|
|
"DeprecationWarning: `old_feature` has been deprecated. Use `new_feature` instead."
|
|
in args[0]
|
|
)
|
|
|
|
|
|
def test_deprecation_warning_error():
|
|
with pytest.raises(ValueError) as excinfo:
|
|
deprecation_warning("old_feature", error=True)
|
|
assert "`old_feature` has been deprecated." in str(excinfo.value)
|
|
|
|
|
|
def test_deprecated_decorator_function():
|
|
with patch("ray._common.deprecation.logger.warning") as mock_warning, patch(
|
|
"ray._common.deprecation.log_once"
|
|
) as mock_log_once:
|
|
mock_log_once.return_value = True
|
|
|
|
@Deprecated(old="old_func", new="new_func", error=False)
|
|
def old_func():
|
|
return "result"
|
|
|
|
result = old_func()
|
|
assert result == "result"
|
|
mock_warning.assert_called_once()
|
|
|
|
|
|
def test_deprecated_decorator_class():
|
|
with patch("ray._common.deprecation.logger.warning") as mock_warning, patch(
|
|
"ray._common.deprecation.log_once"
|
|
) as mock_log_once:
|
|
mock_log_once.return_value = True
|
|
|
|
@Deprecated(old="OldClass", new="NewClass", error=False)
|
|
class OldClass:
|
|
pass
|
|
|
|
instance = OldClass()
|
|
assert isinstance(instance, OldClass)
|
|
mock_warning.assert_called_once()
|
|
|
|
|
|
def test_deprecated_decorator_method():
|
|
with patch("ray._common.deprecation.logger.warning") as mock_warning, patch(
|
|
"ray._common.deprecation.log_once"
|
|
) as mock_log_once:
|
|
mock_log_once.return_value = True
|
|
|
|
class MyClass:
|
|
@Deprecated(old="old_method", new="new_method", error=False)
|
|
def old_method(self):
|
|
return "method_result"
|
|
|
|
instance = MyClass()
|
|
result = instance.old_method()
|
|
assert result == "method_result"
|
|
mock_warning.assert_called_once()
|
|
|
|
|
|
def test_deprecated_decorator_error():
|
|
with patch("ray._common.deprecation.log_once") as mock_log_once:
|
|
mock_log_once.return_value = True
|
|
|
|
@Deprecated(old="old_func", error=True)
|
|
def old_func():
|
|
pass
|
|
|
|
with pytest.raises(ValueError):
|
|
old_func()
|
|
|
|
|
|
def test_deprecated_value_constant():
|
|
assert (
|
|
DEPRECATED_VALUE == -1
|
|
), f"DEPRECATED_VALUE should be -1, but got {DEPRECATED_VALUE}"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-sv", __file__]))
|