Add candletype from string

2021-12-03 12:46:18 +01:00
parent f9cf59bb4d
commit f33643cacf
3 changed files with 26 additions and 0 deletions
--- a/tests/leverage/test_candletype.py
+++ b/tests/leverage/test_candletype.py
@@ -0,0 +1,18 @@
+import pytest
+
+from freqtrade.enums import CandleType
+
+
+@pytest.mark.parametrize('input,expected', [
+    ('', CandleType.SPOT),
+    ('spot', CandleType.SPOT),
+    (CandleType.SPOT, CandleType.SPOT),
+    (CandleType.FUTURES, CandleType.FUTURES),
+    (CandleType.INDEX, CandleType.INDEX),
+    (CandleType.MARK, CandleType.MARK),
+    ('futures', CandleType.FUTURES),
+    ('mark', CandleType.MARK),
+    ('premiumIndex', CandleType.PREMIUMINDEX),
+])
+def test_candle_type_from_string(input, expected):
+    assert CandleType.from_string(input) == expected