From ba8b2fe4cc97dbcfdd226a453ec459a12fc7f4fc Mon Sep 17 00:00:00 2001 From: Adam King Date: Sun, 10 Nov 2019 21:05:13 -0800 Subject: [PATCH] Fix Exchange tests. --- tests/tensortrade/exchanges/test_injection.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/tests/tensortrade/exchanges/test_injection.py b/tests/tensortrade/exchanges/test_injection.py index 88b2dae70..e855128ca 100644 --- a/tests/tensortrade/exchanges/test_injection.py +++ b/tests/tensortrade/exchanges/test_injection.py @@ -6,11 +6,11 @@ from gym import Space from tensortrade import TradingContext -from tensortrade.exchanges import * -from tensortrade.exchanges.live import * -from tensortrade.exchanges.simulated import * from tensortrade.trades import Trade from tensortrade.slippage import SlippageModel +from tensortrade.exchanges import Exchange, get +from tensortrade.exchanges.live import CCXTExchange +from tensortrade.exchanges.simulated import SimulatedExchange, FBMExchange class ConcreteExchange(Exchange): @@ -39,18 +39,14 @@ def performance(self) -> pd.DataFrame: pass @property - def generated_space(self) -> Space: - pass - - @property - def generated_columns(self) -> List[str]: + def observation_columns(self) -> List[str]: pass @property def has_next_observation(self) -> bool: pass - def _create_observation_generator(self) -> Generator[pd.DataFrame, None, None]: + def next_observation(self) -> pd.DataFrame: pass def current_price(self, symbol: str) -> float: