Skip to content

Commit

Permalink
remove insignificant initialization tests
Browse files Browse the repository at this point in the history
  • Loading branch information
jennmueng committed Jul 17, 2024
1 parent bf681ef commit d430b56
Showing 1 changed file with 0 additions and 13 deletions.
13 changes: 0 additions & 13 deletions tests/automation/agent/test_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,14 +31,6 @@ def run_iteration(self):

return TestAgent(config)

def test_initialization(self, agent: LlmAgent, config: AgentConfig):
assert agent.config == config
assert agent.tools == []
assert agent.memory == []
assert isinstance(agent.usage, Usage)
assert agent.name == "Agent"
assert agent.iterations == 0

def test_should_continue(self, agent: LlmAgent):
assert agent.should_continue() # Initial state

Expand Down Expand Up @@ -116,11 +108,6 @@ def agent(self, config):
def mock_client(self):
return resolve(GptClient)

def test_initialization(self, agent, config, mock_client):
assert agent.config == config
assert agent.client == mock_client
assert agent.chat_completion_kwargs == {}

def test_run_iteration(self, agent, mock_client):
mock_message = Message(role="assistant", content="Test response")
mock_usage = Usage(completion_tokens=10, prompt_tokens=20, total_tokens=30)
Expand Down

0 comments on commit d430b56

Please sign in to comment.