load_trades_db should give as many columns as possible

2019-08-03 19:55:54 +02:00
parent fb103dd162
commit c4e30862ee
2 changed files with 21 additions and 5 deletions
--- a/freqtrade/tests/data/test_btanalysis.py
+++ b/freqtrade/tests/data/test_btanalysis.py
@@ -45,6 +45,11 @@ def test_load_trades_db(default_conf, fee, mocker):
    assert isinstance(trades, DataFrame)
    assert "pair" in trades.columns
    assert "open_time" in trades.columns
+    assert "profitperc" in trades.columns
+
+    for col in BT_DATA_COLUMNS:
+        if col not in ['index', 'open_at_end']:
+            assert col in trades.columns


 def test_extract_trades_of_period():