@@ -19,8 +19,8 @@ fn test_swap_total_hotkey_coldkey_stakes_this_interval() {
19
19
let stake = 100 ;
20
20
let block = 42 ;
21
21
22
- OwnedHotkeys :: < Test > :: insert ( & old_coldkey, vec ! [ hotkey] ) ;
23
- TotalHotkeyColdkeyStakesThisInterval :: < Test > :: insert ( & hotkey, & old_coldkey, ( stake, block) ) ;
22
+ OwnedHotkeys :: < Test > :: insert ( old_coldkey, vec ! [ hotkey] ) ;
23
+ TotalHotkeyColdkeyStakesThisInterval :: < Test > :: insert ( hotkey, old_coldkey, ( stake, block) ) ;
24
24
25
25
let mut weight = Weight :: zero ( ) ;
26
26
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -30,11 +30,11 @@ fn test_swap_total_hotkey_coldkey_stakes_this_interval() {
30
30
) ) ;
31
31
32
32
assert ! ( !TotalHotkeyColdkeyStakesThisInterval :: <Test >:: contains_key(
33
- & hotkey,
34
- & old_coldkey
33
+ hotkey,
34
+ old_coldkey
35
35
) ) ;
36
36
assert_eq ! (
37
- TotalHotkeyColdkeyStakesThisInterval :: <Test >:: get( & hotkey, & new_coldkey) ,
37
+ TotalHotkeyColdkeyStakesThisInterval :: <Test >:: get( hotkey, new_coldkey) ,
38
38
( stake, block)
39
39
) ;
40
40
} ) ;
@@ -71,17 +71,17 @@ fn test_swap_stake() {
71
71
let hotkey = U256 :: from ( 3 ) ;
72
72
let stake = 100 ;
73
73
74
- StakingHotkeys :: < Test > :: insert ( & old_coldkey, vec ! [ hotkey] ) ;
75
- Stake :: < Test > :: insert ( & hotkey, & old_coldkey, stake) ;
74
+ StakingHotkeys :: < Test > :: insert ( old_coldkey, vec ! [ hotkey] ) ;
75
+ Stake :: < Test > :: insert ( hotkey, old_coldkey, stake) ;
76
76
let mut weight = Weight :: zero ( ) ;
77
77
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
78
78
& old_coldkey,
79
79
& new_coldkey,
80
80
& mut weight
81
81
) ) ;
82
82
83
- assert ! ( !Stake :: <Test >:: contains_key( & hotkey, & old_coldkey) ) ;
84
- assert_eq ! ( Stake :: <Test >:: get( & hotkey, & new_coldkey) , stake) ;
83
+ assert ! ( !Stake :: <Test >:: contains_key( hotkey, old_coldkey) ) ;
84
+ assert_eq ! ( Stake :: <Test >:: get( hotkey, new_coldkey) , stake) ;
85
85
} ) ;
86
86
}
87
87
@@ -93,7 +93,7 @@ fn test_swap_total_coldkey_stake() {
93
93
let new_coldkey = U256 :: from ( 2 ) ;
94
94
let stake = 100 ;
95
95
96
- TotalColdkeyStake :: < Test > :: insert ( & old_coldkey, stake) ;
96
+ TotalColdkeyStake :: < Test > :: insert ( old_coldkey, stake) ;
97
97
98
98
let mut weight = Weight :: zero ( ) ;
99
99
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -102,8 +102,8 @@ fn test_swap_total_coldkey_stake() {
102
102
& mut weight
103
103
) ) ;
104
104
105
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & old_coldkey) , 0 ) ;
106
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & new_coldkey) , stake) ;
105
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( old_coldkey) , 0 ) ;
106
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( new_coldkey) , stake) ;
107
107
} ) ;
108
108
}
109
109
@@ -115,7 +115,7 @@ fn test_swap_staking_hotkeys() {
115
115
let new_coldkey = U256 :: from ( 2 ) ;
116
116
let hotkey = U256 :: from ( 3 ) ;
117
117
118
- StakingHotkeys :: < Test > :: insert ( & old_coldkey, vec ! [ hotkey] ) ;
118
+ StakingHotkeys :: < Test > :: insert ( old_coldkey, vec ! [ hotkey] ) ;
119
119
120
120
let mut weight = Weight :: zero ( ) ;
121
121
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -124,8 +124,8 @@ fn test_swap_staking_hotkeys() {
124
124
& mut weight
125
125
) ) ;
126
126
127
- assert ! ( StakingHotkeys :: <Test >:: get( & old_coldkey) . is_empty( ) ) ;
128
- assert_eq ! ( StakingHotkeys :: <Test >:: get( & new_coldkey) , vec![ hotkey] ) ;
127
+ assert ! ( StakingHotkeys :: <Test >:: get( old_coldkey) . is_empty( ) ) ;
128
+ assert_eq ! ( StakingHotkeys :: <Test >:: get( new_coldkey) , vec![ hotkey] ) ;
129
129
} ) ;
130
130
}
131
131
@@ -137,8 +137,8 @@ fn test_swap_hotkey_owners() {
137
137
let new_coldkey = U256 :: from ( 2 ) ;
138
138
let hotkey = U256 :: from ( 3 ) ;
139
139
140
- Owner :: < Test > :: insert ( & hotkey, & old_coldkey) ;
141
- OwnedHotkeys :: < Test > :: insert ( & old_coldkey, vec ! [ hotkey] ) ;
140
+ Owner :: < Test > :: insert ( hotkey, old_coldkey) ;
141
+ OwnedHotkeys :: < Test > :: insert ( old_coldkey, vec ! [ hotkey] ) ;
142
142
143
143
let mut weight = Weight :: zero ( ) ;
144
144
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -147,9 +147,9 @@ fn test_swap_hotkey_owners() {
147
147
& mut weight
148
148
) ) ;
149
149
150
- assert_eq ! ( Owner :: <Test >:: get( & hotkey) , new_coldkey) ;
151
- assert ! ( OwnedHotkeys :: <Test >:: get( & old_coldkey) . is_empty( ) ) ;
152
- assert_eq ! ( OwnedHotkeys :: <Test >:: get( & new_coldkey) , vec![ hotkey] ) ;
150
+ assert_eq ! ( Owner :: <Test >:: get( hotkey) , new_coldkey) ;
151
+ assert ! ( OwnedHotkeys :: <Test >:: get( old_coldkey) . is_empty( ) ) ;
152
+ assert_eq ! ( OwnedHotkeys :: <Test >:: get( new_coldkey) , vec![ hotkey] ) ;
153
153
} ) ;
154
154
}
155
155
// SKIP_WASM_BUILD=1 RUST_LOG=debug cargo test --test swap_coldkey -- test_transfer_remaining_balance --exact --nocapture
@@ -188,8 +188,8 @@ fn test_swap_with_no_stake() {
188
188
& mut weight
189
189
) ) ;
190
190
191
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & old_coldkey) , 0 ) ;
192
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & new_coldkey) , 0 ) ;
191
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( old_coldkey) , 0 ) ;
192
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( new_coldkey) , 0 ) ;
193
193
} ) ;
194
194
}
195
195
@@ -202,7 +202,7 @@ fn test_swap_with_multiple_hotkeys() {
202
202
let hotkey1 = U256 :: from ( 3 ) ;
203
203
let hotkey2 = U256 :: from ( 4 ) ;
204
204
205
- OwnedHotkeys :: < Test > :: insert ( & old_coldkey, vec ! [ hotkey1, hotkey2] ) ;
205
+ OwnedHotkeys :: < Test > :: insert ( old_coldkey, vec ! [ hotkey1, hotkey2] ) ;
206
206
207
207
let mut weight = Weight :: zero ( ) ;
208
208
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -211,9 +211,9 @@ fn test_swap_with_multiple_hotkeys() {
211
211
& mut weight
212
212
) ) ;
213
213
214
- assert ! ( OwnedHotkeys :: <Test >:: get( & old_coldkey) . is_empty( ) ) ;
214
+ assert ! ( OwnedHotkeys :: <Test >:: get( old_coldkey) . is_empty( ) ) ;
215
215
assert_eq ! (
216
- OwnedHotkeys :: <Test >:: get( & new_coldkey) ,
216
+ OwnedHotkeys :: <Test >:: get( new_coldkey) ,
217
217
vec![ hotkey1, hotkey2]
218
218
) ;
219
219
} ) ;
@@ -230,8 +230,8 @@ fn test_swap_with_multiple_subnets() {
230
230
231
231
add_network ( netuid1, 1 , 0 ) ;
232
232
add_network ( netuid2, 1 , 0 ) ;
233
- SubnetOwner :: < Test > :: insert ( netuid1, & old_coldkey) ;
234
- SubnetOwner :: < Test > :: insert ( netuid2, & old_coldkey) ;
233
+ SubnetOwner :: < Test > :: insert ( netuid1, old_coldkey) ;
234
+ SubnetOwner :: < Test > :: insert ( netuid2, old_coldkey) ;
235
235
236
236
let mut weight = Weight :: zero ( ) ;
237
237
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -259,8 +259,8 @@ fn test_swap_with_zero_balance() {
259
259
& mut weight
260
260
) ) ;
261
261
262
- assert_eq ! ( Balances :: free_balance( & old_coldkey) , 0 ) ;
263
- assert_eq ! ( Balances :: free_balance( & new_coldkey) , 0 ) ;
262
+ assert_eq ! ( Balances :: free_balance( old_coldkey) , 0 ) ;
263
+ assert_eq ! ( Balances :: free_balance( new_coldkey) , 0 ) ;
264
264
} ) ;
265
265
}
266
266
@@ -272,7 +272,7 @@ fn test_swap_idempotency() {
272
272
let new_coldkey = U256 :: from ( 2 ) ;
273
273
let stake = 100 ;
274
274
275
- TotalColdkeyStake :: < Test > :: insert ( & old_coldkey, stake) ;
275
+ TotalColdkeyStake :: < Test > :: insert ( old_coldkey, stake) ;
276
276
277
277
let mut weight = Weight :: zero ( ) ;
278
278
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -286,8 +286,8 @@ fn test_swap_idempotency() {
286
286
& mut weight
287
287
) ) ;
288
288
289
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & old_coldkey) , 0 ) ;
290
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & new_coldkey) , stake) ;
289
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( old_coldkey) , 0 ) ;
290
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( new_coldkey) , stake) ;
291
291
} ) ;
292
292
}
293
293
@@ -299,7 +299,7 @@ fn test_swap_with_max_values() {
299
299
let new_coldkey = U256 :: from ( 2 ) ;
300
300
let max_stake = u64:: MAX ;
301
301
302
- TotalColdkeyStake :: < Test > :: insert ( & old_coldkey, max_stake) ;
302
+ TotalColdkeyStake :: < Test > :: insert ( old_coldkey, max_stake) ;
303
303
304
304
let mut weight = Weight :: zero ( ) ;
305
305
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -308,8 +308,8 @@ fn test_swap_with_max_values() {
308
308
& mut weight
309
309
) ) ;
310
310
311
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & old_coldkey) , 0 ) ;
312
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & new_coldkey) , max_stake) ;
311
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( old_coldkey) , 0 ) ;
312
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( new_coldkey) , max_stake) ;
313
313
} ) ;
314
314
}
315
315
@@ -321,7 +321,7 @@ fn test_swap_with_non_existent_new_coldkey() {
321
321
let new_coldkey = U256 :: from ( 2 ) ;
322
322
let stake = 100 ;
323
323
324
- TotalColdkeyStake :: < Test > :: insert ( & old_coldkey, stake) ;
324
+ TotalColdkeyStake :: < Test > :: insert ( old_coldkey, stake) ;
325
325
326
326
let mut weight = Weight :: zero ( ) ;
327
327
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -330,8 +330,8 @@ fn test_swap_with_non_existent_new_coldkey() {
330
330
& mut weight
331
331
) ) ;
332
332
333
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & old_coldkey) , 0 ) ;
334
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & new_coldkey) , stake) ;
333
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( old_coldkey) , 0 ) ;
334
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( new_coldkey) , stake) ;
335
335
} ) ;
336
336
}
337
337
@@ -343,8 +343,8 @@ fn test_swap_with_overflow_in_stake_addition() {
343
343
let new_coldkey = U256 :: from ( 2 ) ;
344
344
let max_stake = u64:: MAX ;
345
345
346
- TotalColdkeyStake :: < Test > :: insert ( & old_coldkey, max_stake) ;
347
- TotalColdkeyStake :: < Test > :: insert ( & new_coldkey, 1 ) ;
346
+ TotalColdkeyStake :: < Test > :: insert ( old_coldkey, max_stake) ;
347
+ TotalColdkeyStake :: < Test > :: insert ( new_coldkey, 1 ) ;
348
348
349
349
let mut weight = Weight :: zero ( ) ;
350
350
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -353,8 +353,8 @@ fn test_swap_with_overflow_in_stake_addition() {
353
353
& mut weight
354
354
) ) ;
355
355
356
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & old_coldkey) , 0 ) ;
357
- assert_eq ! ( TotalColdkeyStake :: <Test >:: get( & new_coldkey) , max_stake) ;
356
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( old_coldkey) , 0 ) ;
357
+ assert_eq ! ( TotalColdkeyStake :: <Test >:: get( new_coldkey) , max_stake) ;
358
358
} ) ;
359
359
}
360
360
@@ -367,7 +367,7 @@ fn test_swap_with_max_hotkeys() {
367
367
let max_hotkeys = 1000 ;
368
368
let hotkeys: Vec < U256 > = ( 0 ..max_hotkeys) . map ( U256 :: from) . collect ( ) ;
369
369
370
- OwnedHotkeys :: < Test > :: insert ( & old_coldkey, hotkeys. clone ( ) ) ;
370
+ OwnedHotkeys :: < Test > :: insert ( old_coldkey, hotkeys. clone ( ) ) ;
371
371
372
372
let mut weight = Weight :: zero ( ) ;
373
373
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -376,8 +376,8 @@ fn test_swap_with_max_hotkeys() {
376
376
& mut weight
377
377
) ) ;
378
378
379
- assert ! ( OwnedHotkeys :: <Test >:: get( & old_coldkey) . is_empty( ) ) ;
380
- assert_eq ! ( OwnedHotkeys :: <Test >:: get( & new_coldkey) , hotkeys) ;
379
+ assert ! ( OwnedHotkeys :: <Test >:: get( old_coldkey) . is_empty( ) ) ;
380
+ assert_eq ! ( OwnedHotkeys :: <Test >:: get( new_coldkey) , hotkeys) ;
381
381
} ) ;
382
382
}
383
383
@@ -391,10 +391,10 @@ fn test_swap_effect_on_delegated_stake() {
391
391
let hotkey = U256 :: from ( 4 ) ;
392
392
let stake = 100 ;
393
393
394
- StakingHotkeys :: < Test > :: insert ( & old_coldkey, vec ! [ hotkey] ) ;
395
- StakingHotkeys :: < Test > :: insert ( & delegator, vec ! [ hotkey] ) ;
396
- Stake :: < Test > :: insert ( & hotkey, & old_coldkey, stake) ;
397
- Stake :: < Test > :: insert ( & hotkey, & delegator, stake) ;
394
+ StakingHotkeys :: < Test > :: insert ( old_coldkey, vec ! [ hotkey] ) ;
395
+ StakingHotkeys :: < Test > :: insert ( delegator, vec ! [ hotkey] ) ;
396
+ Stake :: < Test > :: insert ( hotkey, old_coldkey, stake) ;
397
+ Stake :: < Test > :: insert ( hotkey, delegator, stake) ;
398
398
399
399
let mut weight = Weight :: zero ( ) ;
400
400
assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -403,9 +403,9 @@ fn test_swap_effect_on_delegated_stake() {
403
403
& mut weight
404
404
) ) ;
405
405
406
- assert_eq ! ( Stake :: <Test >:: get( & hotkey, & new_coldkey) , stake) ;
407
- assert_eq ! ( Stake :: <Test >:: get( & hotkey, & delegator) , stake) ;
408
- assert_eq ! ( Stake :: <Test >:: get( & hotkey, & old_coldkey) , 0 ) ;
406
+ assert_eq ! ( Stake :: <Test >:: get( hotkey, new_coldkey) , stake) ;
407
+ assert_eq ! ( Stake :: <Test >:: get( hotkey, delegator) , stake) ;
408
+ assert_eq ! ( Stake :: <Test >:: get( hotkey, old_coldkey) , 0 ) ;
409
409
} ) ;
410
410
}
411
411
@@ -420,8 +420,8 @@ fn test_swap_concurrent_modifications() {
420
420
let initial_stake = 100 ;
421
421
let additional_stake = 50 ;
422
422
423
- StakingHotkeys :: < Test > :: insert ( & old_coldkey, vec ! [ hotkey] ) ;
424
- Stake :: < Test > :: insert ( & hotkey, & old_coldkey, initial_stake) ;
423
+ StakingHotkeys :: < Test > :: insert ( old_coldkey, vec ! [ hotkey] ) ;
424
+ Stake :: < Test > :: insert ( hotkey, old_coldkey, initial_stake) ;
425
425
426
426
// Simulate concurrent stake addition
427
427
add_network ( netuid, 1 , 1 ) ;
@@ -441,10 +441,10 @@ fn test_swap_concurrent_modifications() {
441
441
) ) ;
442
442
443
443
assert_eq ! (
444
- Stake :: <Test >:: get( & hotkey, & new_coldkey) ,
444
+ Stake :: <Test >:: get( hotkey, new_coldkey) ,
445
445
initial_stake + additional_stake - 1
446
446
) ;
447
- assert ! ( !Stake :: <Test >:: contains_key( & hotkey, & old_coldkey) ) ;
447
+ assert ! ( !Stake :: <Test >:: contains_key( hotkey, old_coldkey) ) ;
448
448
} ) ;
449
449
}
450
450
0 commit comments