Fix some tests

2021-08-24 06:45:09 +02:00
parent 7a977a8eaf
commit 11bd8e912e
6 changed files with 44 additions and 27 deletions
--- a/tests/optimize/init.py
+++ b/tests/optimize/init.py
@@ -56,6 +56,8 @@ def _build_backtest_dataframe(data):
    # Ensure floats are in place
    for column in ['open', 'high', 'low', 'close', 'volume']:
        frame[column] = frame[column].astype('float64')
-    if 'buy_tag' not in columns:
-        frame['buy_tag'] = None
+    if 'long_tag' not in columns:
+        frame['long_tag'] = None
+    if 'short_tag' not in columns:
+        frame['short_tag'] = None
    return frame