From ced3670302ad71cef7cbe6d4e01d21d2d7265594 Mon Sep 17 00:00:00 2001 From: Masanori HIRANO Date: Tue, 18 Apr 2023 01:02:39 +0900 Subject: [PATCH] fix --- tests/pams/events/test_fundamental_price_shock.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/pams/events/test_fundamental_price_shock.py b/tests/pams/events/test_fundamental_price_shock.py index a83d679d..758be542 100644 --- a/tests/pams/events/test_fundamental_price_shock.py +++ b/tests/pams/events/test_fundamental_price_shock.py @@ -5,13 +5,14 @@ from pams import Market from pams import Session from pams import Simulator +from pams.events import EventABC from pams.events import FundamentalPriceShock from pams.logs import Logger from tests.pams.events.test_base import TestEventABC class TestFundamentalPriceShock(TestEventABC): - def test__init__(self) -> None: + def test__init__(self) -> EventABC: sim = Simulator(prng=random.Random(4)) logger = Logger() session = Session( @@ -163,6 +164,7 @@ def test__init__(self) -> None: } with pytest.raises(ValueError): event.setup(settings=setting9) + return event def test_hook_registration(self) -> None: sim = Simulator(prng=random.Random(4))