forked from cleveragents/cleveragents-core
35 lines
1012 B
Python
35 lines
1012 B
Python
from behave import given
|
|
from behave import then
|
|
from behave import when
|
|
|
|
|
|
@given("I import the agent module")
|
|
def step_impl(context):
|
|
import cleveragents.agent
|
|
|
|
context.agent_module = cleveragents.agent
|
|
|
|
|
|
@then("I should be able to access the Agent class")
|
|
def step_impl(context):
|
|
assert hasattr(
|
|
context.agent_module, "Agent"
|
|
), "Agent class not found in agent module"
|
|
from cleveragents.agents.base import Agent as OriginalAgent
|
|
|
|
assert (
|
|
context.agent_module.Agent is OriginalAgent
|
|
), "Agent class is not the expected class"
|
|
|
|
|
|
@then("I should be able to access the AgentWithMemory class")
|
|
def step_impl(context):
|
|
assert hasattr(
|
|
context.agent_module, "AgentWithMemory"
|
|
), "AgentWithMemory class not found in agent module"
|
|
from cleveragents.agents.base import AgentWithMemory as OriginalAgentWithMemory
|
|
|
|
assert (
|
|
context.agent_module.AgentWithMemory is OriginalAgentWithMemory
|
|
), "AgentWithMemory class is not the expected class"
|