fix failing test

AsyncAlgoTrading · May 20, 2019 · b9d3dc6 · b9d3dc6
1 parent bc7325a
commit b9d3dc6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/aat/tests/test_backtest.py b/aat/tests/test_backtest.py
@@ -5,13 +5,14 @@ class TestBacktest:
     def setup(self):
         from ..config import BacktestConfig
         self.config = BacktestConfig()
-        df = pd.DataFrame([{'volume': '100',
-                            'close': '1',
+        df = pd.DataFrame([{'volume': 100.0,
+                            'close': 1.0,
                             'timestamp': '1558296780000',
+                            'exchange': 'GEMINI',
                             'pair': 'USDBTC'
                             }])
         df['timestamp'] = pd.to_datetime(df['timestamp'], unit='ms')
-        df.set_index(['timestamp', 'pair'])
+        df.set_index(['timestamp', 'pair'], inplace=True)
         self.test_line = df.iloc[0]
 
     def teardown(self):