65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
|
import pytest
|
||
|
|
from src.agent_app import CreditCardFraudAgent, create_agent, Tool
|
||
|
|
|
||
|
|
|
||
|
|
def test_agent_initialization():
|
||
|
|
agent = CreditCardFraudAgent(model_dir="models", model_name="random_forest")
|
||
|
|
assert agent.inference is not None
|
||
|
|
assert len(agent.tools) == 2
|
||
|
|
|
||
|
|
|
||
|
|
def test_create_agent():
|
||
|
|
agent = create_agent(model_dir="models", model_name="random_forest")
|
||
|
|
assert isinstance(agent, CreditCardFraudAgent)
|
||
|
|
|
||
|
|
|
||
|
|
def test_list_tools():
|
||
|
|
agent = create_agent()
|
||
|
|
tools = agent.list_tools()
|
||
|
|
assert len(tools) == 2
|
||
|
|
tool_names = [tool["name"] for tool in tools]
|
||
|
|
assert "predict_fraud" in tool_names
|
||
|
|
assert "analyze_transaction" in tool_names
|
||
|
|
|
||
|
|
|
||
|
|
def test_tool_structure():
|
||
|
|
agent = create_agent()
|
||
|
|
for tool in agent.tools:
|
||
|
|
assert hasattr(tool, "name")
|
||
|
|
assert hasattr(tool, "description")
|
||
|
|
assert hasattr(tool, "func")
|
||
|
|
assert callable(tool.func)
|
||
|
|
|
||
|
|
|
||
|
|
def test_analyze_transaction():
|
||
|
|
agent = create_agent()
|
||
|
|
transaction = [
|
||
|
|
0, -1.36, -0.07, 2.54, 1.38, -0.34, 0.46, 0.24, 0.10, 0.36,
|
||
|
|
0.09, -0.55, -0.62, -0.99, -0.31, 1.47, -0.47, 0.21, 0.03, 0.40,
|
||
|
|
0.25, -0.02, 0.28, -0.11, 0.07, 0.13, -0.19, 0.13, -0.02, 149.62
|
||
|
|
]
|
||
|
|
analysis = agent._analyze_transaction(transaction)
|
||
|
|
assert "feature_count" in analysis
|
||
|
|
assert "amount" in analysis
|
||
|
|
assert "time" in analysis
|
||
|
|
assert "statistics" in analysis
|
||
|
|
assert "anomalies" in analysis
|
||
|
|
assert analysis["feature_count"] == 30
|
||
|
|
assert analysis["amount"] == 149.62
|
||
|
|
|
||
|
|
|
||
|
|
def test_process_transaction():
|
||
|
|
agent = create_agent()
|
||
|
|
transaction = [
|
||
|
|
0, -1.36, -0.07, 2.54, 1.38, -0.34, 0.46, 0.24, 0.10, 0.36,
|
||
|
|
0.09, -0.55, -0.62, -0.99, -0.31, 1.47, -0.47, 0.21, 0.03, 0.40,
|
||
|
|
0.25, -0.02, 0.28, -0.11, 0.07, 0.13, -0.19, 0.13, -0.02, 149.62
|
||
|
|
]
|
||
|
|
result = agent.process_transaction(transaction)
|
||
|
|
assert result.evaluation is not None
|
||
|
|
assert result.explanation is not None
|
||
|
|
assert result.action_plan is not None
|
||
|
|
assert hasattr(result.evaluation, "predicted_class")
|
||
|
|
assert hasattr(result.evaluation, "fraud_probability")
|
||
|
|
assert hasattr(result.action_plan, "actions")
|