fix tests

AsyncAlgoTrading · Jul 2, 2019 · b70ff16 · b70ff16
1 parent e2d32c9
commit b70ff16
Show file tree

Hide file tree

Showing 12 changed files with 5 additions and 667 deletions.
diff --git a/aat/strategies/data_capture.py b/aat/strategies/data_capture.py
@@ -8,12 +8,6 @@ def __init__(self, filename) -> None:
         super(DataCaptureStrategy, self).__init__()
         self.filename = filename
 
-    def onBuy(self, res: TradeResponse) -> None:
-        log.info(res)
-
-    def onSell(self, res: TradeResponse) -> None:
-        log.info(res)
-
     def onTrade(self, data: MarketData) -> bool:
         log.info(data)
 
@@ -26,8 +20,8 @@ def onExit(self) -> None:
     def onChange(self, data: MarketData) -> None:
         log.info(data)
 
-    def onFill(self, data: MarketData) -> None:
-        log.info(data)
+    def onFill(self, res: TradeResponse) -> None:
+        log.info(res)
 
     def onCancel(self, data: MarketData) -> None:
         log.info(data)

diff --git a/aat/strategies/sma_crosses_strategy.py b/aat/strategies/sma_crosses_strategy.py
diff --git a/aat/strategies/test_strat.py b/aat/strategies/test_strat.py
diff --git a/aat/tests/__init__.py b/aat/tests/__init__.py
@@ -1,3 +0,0 @@
-from .exchanges import test_coinbase
-from .exchanges import test_gemini
-from .strategies import test_sma_crosses_strategy

diff --git a/aat/tests/exchanges/test_coinbase.py b/aat/tests/exchanges/test_coinbase.py
@@ -116,7 +116,7 @@ def test_CoinbaseHelpers_strToTradeType(self):
         assert e.strToTradeType('match') == TickType.TRADE
         assert e.strToTradeType('open') == TickType.OPEN
         assert e.strToTradeType('change', 'canceled') == TickType.CANCEL
-        assert e.strToTradeType('change', 'filled') == TickType.FILL
+        assert e.strToTradeType('done', 'filled') == TickType.TRADE
         assert e.strToTradeType('change') == TickType.CHANGE
         assert e.strToTradeType('heartbeat') == TickType.HEARTBEAT
         assert e.strToTradeType('flarg') == TickType.ERROR