Skip to content

Commit

Permalink
add test
Browse files Browse the repository at this point in the history
  • Loading branch information
masanorihirano committed Oct 25, 2024
1 parent abd39a4 commit 33f8613
Showing 1 changed file with 46 additions and 0 deletions.
46 changes: 46 additions & 0 deletions tests/pams/runners/test_sequential.py
Original file line number Diff line number Diff line change
Expand Up @@ -1559,3 +1559,49 @@ def process_market_step_end_log(self, log: MarketStepEndLog) -> None:
assert logger.n_market_step_end == sum(
[session.iteration_steps for session in runner.simulator.sessions]
)

def test_collect_orders_from_normal_agents_error_1(self) -> None:
runner = self.test__init__(
setting_mode="dict", logger=None, simulator_class=None
)
runner._setup()

dummy_order = Order(
agent_id=100,
market_id=2,
is_buy=True,
kind=LIMIT_ORDER,
volume=1,
price=300.0,
)

with mock.patch(
"pams.agents.fcn_agent.FCNAgent.submit_orders", return_value=[dummy_order]
):
with pytest.raises(ValueError):
_ = runner._collect_orders_from_normal_agents(
session=runner.simulator.sessions[0]
)

setting = copy.deepcopy(self.default_setting)
setting["simulation"]["sessions"][0]["withOrderPlacement"] = False # type: ignore
runner = self.test__init__(
setting_mode="dict", logger=None, simulator_class=None, setting=setting
)
runner._setup()
dummy_order = Order(
agent_id=100,
market_id=2,
is_buy=True,
kind=LIMIT_ORDER,
volume=1,
price=300.0,
)

with mock.patch(
"pams.agents.fcn_agent.FCNAgent.submit_orders", return_value=[dummy_order]
):
with pytest.raises(AssertionError):
_ = runner._collect_orders_from_normal_agents(
session=runner.simulator.sessions[0]
)

0 comments on commit 33f8613

Please sign in to comment.