You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
982 B
33 lines
982 B
from swarms.agents.exceptions import (
|
|
ErrorSeverity,
|
|
ToolAgentError,
|
|
ValidationError,
|
|
ModelNotProvidedError
|
|
)
|
|
|
|
def test_error_severity():
|
|
assert ErrorSeverity.LOW.value == "low"
|
|
assert ErrorSeverity.MEDIUM.value == "medium"
|
|
assert ErrorSeverity.HIGH.value == "high"
|
|
assert ErrorSeverity.CRITICAL.value == "critical"
|
|
|
|
def test_tool_agent_error():
|
|
error = ToolAgentError(
|
|
"Test error",
|
|
severity=ErrorSeverity.HIGH,
|
|
details={"test": "value"}
|
|
)
|
|
assert str(error) == "Test error"
|
|
assert error.severity == ErrorSeverity.HIGH
|
|
assert error.details == {"test": "value"}
|
|
|
|
def test_validation_error():
|
|
error = ValidationError("Validation failed")
|
|
assert isinstance(error, ToolAgentError)
|
|
assert str(error) == "Validation failed"
|
|
|
|
def test_model_not_provided_error():
|
|
error = ModelNotProvidedError("Model missing")
|
|
assert isinstance(error, ToolAgentError)
|
|
assert str(error) == "Model missing"
|