Explicit test for candletype get_default

2021-12-08 16:20:26 +01:00
parent 35afc7b478
commit 2679744228
2 changed files with 10 additions and 2 deletions
--- a/tests/leverage/test_candletype.py
+++ b/tests/leverage/test_candletype.py
@@ -14,5 +14,14 @@ from freqtrade.enums import CandleType
    ('mark', CandleType.MARK),
    ('premiumIndex', CandleType.PREMIUMINDEX),
 ])
-def test_candle_type_from_string(input, expected):
+def test_CandleType_from_string(input, expected):
    assert CandleType.from_string(input) == expected
+
+
+@pytest.mark.parametrize('input,expected', [
+    ('futures', CandleType.FUTURES),
+    ('spot', CandleType.SPOT),
+    ('margin', CandleType.SPOT),
+])
+def test_CandleType_get_default(input, expected):
+    assert CandleType.get_default(input) == expected