Skip to content

Commit 77e1d46

Browse files
committed
Fix tests for new models
1 parent b3057ca commit 77e1d46

File tree

1 file changed

+16
-6
lines changed

1 file changed

+16
-6
lines changed

tests/test_strategies.py

Lines changed: 16 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,9 @@ def mock_discover_strategies():
8888
),
8989
parameters={
9090
"stop_loss": StrategyParameter(
91-
name="Stop Loss",
91+
name="stop_loss",
92+
pretty_name="Stop Loss",
93+
description="Stop loss percentage",
9294
group="Risk Management",
9395
type="Decimal",
9496
prompt="Enter stop loss value",
@@ -98,7 +100,9 @@ def mock_discover_strategies():
98100
max_value=Decimal("1")
99101
),
100102
"take_profit": StrategyParameter(
101-
name="Take Profit",
103+
name="take_profit",
104+
pretty_name="Take Profit",
105+
description="Take profit percentage",
102106
group="Risk Management",
103107
type="Decimal",
104108
prompt="Enter take profit value",
@@ -108,7 +112,9 @@ def mock_discover_strategies():
108112
max_value=Decimal("1")
109113
),
110114
"time_limit": StrategyParameter(
111-
name="Time Limit",
115+
name="time_limit",
116+
pretty_name="Time Limit",
117+
description="Time limit in seconds",
112118
group="General Settings",
113119
type="int",
114120
prompt="Enter time limit in seconds",
@@ -117,7 +123,9 @@ def mock_discover_strategies():
117123
min_value=0
118124
),
119125
"leverage": StrategyParameter(
120-
name="Leverage",
126+
name="leverage",
127+
pretty_name="Leverage",
128+
description="Leverage multiplier",
121129
group="Risk Management",
122130
type="int",
123131
prompt="Enter leverage multiplier",
@@ -127,7 +135,9 @@ def mock_discover_strategies():
127135
is_advanced=True
128136
),
129137
"trading_pair": StrategyParameter(
130-
name="Trading Pair",
138+
name="trading_pair",
139+
pretty_name="Trading Pair",
140+
description="Trading pair to use",
131141
group="General Settings",
132142
type="str",
133143
prompt="Enter trading pair",
@@ -163,7 +173,7 @@ def test_convert_to_strategy_parameter():
163173
field = MockStrategyConfig.__fields__["stop_loss"]
164174
param = convert_to_strategy_parameter("stop_loss", field)
165175

166-
assert param.name == "Stop Loss"
176+
assert param.pretty_name == "Stop Loss"
167177
assert param.group == "Risk Management"
168178
assert param.type == "ConstrainedDecimalValue" # We want the base type, not the constrained type
169179
assert param.default == Decimal("0.03")

0 commit comments

Comments
 (0)