@@ -19,8 +19,8 @@ fn test_swap_total_hotkey_coldkey_stakes_this_interval() {
1919 let stake = 100 ;
2020 let block = 42 ;
2121
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) ) ;
2424
2525 let mut weight = Weight :: zero ( ) ;
2626 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -30,11 +30,11 @@ fn test_swap_total_hotkey_coldkey_stakes_this_interval() {
3030 ) ) ;
3131
3232 assert ! ( !TotalHotkeyColdkeyStakesThisInterval :: <Test >:: contains_key(
33- & hotkey,
34- & old_coldkey
33+ hotkey,
34+ old_coldkey
3535 ) ) ;
3636 assert_eq ! (
37- TotalHotkeyColdkeyStakesThisInterval :: <Test >:: get( & hotkey, & new_coldkey) ,
37+ TotalHotkeyColdkeyStakesThisInterval :: <Test >:: get( hotkey, new_coldkey) ,
3838 ( stake, block)
3939 ) ;
4040 } ) ;
@@ -71,17 +71,17 @@ fn test_swap_stake() {
7171 let hotkey = U256 :: from ( 3 ) ;
7272 let stake = 100 ;
7373
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) ;
7676 let mut weight = Weight :: zero ( ) ;
7777 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
7878 & old_coldkey,
7979 & new_coldkey,
8080 & mut weight
8181 ) ) ;
8282
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) ;
8585 } ) ;
8686}
8787
@@ -93,7 +93,7 @@ fn test_swap_total_coldkey_stake() {
9393 let new_coldkey = U256 :: from ( 2 ) ;
9494 let stake = 100 ;
9595
96- TotalColdkeyStake :: < Test > :: insert ( & old_coldkey, stake) ;
96+ TotalColdkeyStake :: < Test > :: insert ( old_coldkey, stake) ;
9797
9898 let mut weight = Weight :: zero ( ) ;
9999 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -102,8 +102,8 @@ fn test_swap_total_coldkey_stake() {
102102 & mut weight
103103 ) ) ;
104104
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) ;
107107 } ) ;
108108}
109109
@@ -115,7 +115,7 @@ fn test_swap_staking_hotkeys() {
115115 let new_coldkey = U256 :: from ( 2 ) ;
116116 let hotkey = U256 :: from ( 3 ) ;
117117
118- StakingHotkeys :: < Test > :: insert ( & old_coldkey, vec ! [ hotkey] ) ;
118+ StakingHotkeys :: < Test > :: insert ( old_coldkey, vec ! [ hotkey] ) ;
119119
120120 let mut weight = Weight :: zero ( ) ;
121121 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -124,8 +124,8 @@ fn test_swap_staking_hotkeys() {
124124 & mut weight
125125 ) ) ;
126126
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] ) ;
129129 } ) ;
130130}
131131
@@ -137,8 +137,8 @@ fn test_swap_hotkey_owners() {
137137 let new_coldkey = U256 :: from ( 2 ) ;
138138 let hotkey = U256 :: from ( 3 ) ;
139139
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] ) ;
142142
143143 let mut weight = Weight :: zero ( ) ;
144144 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -147,9 +147,9 @@ fn test_swap_hotkey_owners() {
147147 & mut weight
148148 ) ) ;
149149
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] ) ;
153153 } ) ;
154154}
155155// 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() {
188188 & mut weight
189189 ) ) ;
190190
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 ) ;
193193 } ) ;
194194}
195195
@@ -202,7 +202,7 @@ fn test_swap_with_multiple_hotkeys() {
202202 let hotkey1 = U256 :: from ( 3 ) ;
203203 let hotkey2 = U256 :: from ( 4 ) ;
204204
205- OwnedHotkeys :: < Test > :: insert ( & old_coldkey, vec ! [ hotkey1, hotkey2] ) ;
205+ OwnedHotkeys :: < Test > :: insert ( old_coldkey, vec ! [ hotkey1, hotkey2] ) ;
206206
207207 let mut weight = Weight :: zero ( ) ;
208208 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -211,9 +211,9 @@ fn test_swap_with_multiple_hotkeys() {
211211 & mut weight
212212 ) ) ;
213213
214- assert ! ( OwnedHotkeys :: <Test >:: get( & old_coldkey) . is_empty( ) ) ;
214+ assert ! ( OwnedHotkeys :: <Test >:: get( old_coldkey) . is_empty( ) ) ;
215215 assert_eq ! (
216- OwnedHotkeys :: <Test >:: get( & new_coldkey) ,
216+ OwnedHotkeys :: <Test >:: get( new_coldkey) ,
217217 vec![ hotkey1, hotkey2]
218218 ) ;
219219 } ) ;
@@ -230,8 +230,8 @@ fn test_swap_with_multiple_subnets() {
230230
231231 add_network ( netuid1, 1 , 0 ) ;
232232 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) ;
235235
236236 let mut weight = Weight :: zero ( ) ;
237237 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -259,8 +259,8 @@ fn test_swap_with_zero_balance() {
259259 & mut weight
260260 ) ) ;
261261
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 ) ;
264264 } ) ;
265265}
266266
@@ -272,7 +272,7 @@ fn test_swap_idempotency() {
272272 let new_coldkey = U256 :: from ( 2 ) ;
273273 let stake = 100 ;
274274
275- TotalColdkeyStake :: < Test > :: insert ( & old_coldkey, stake) ;
275+ TotalColdkeyStake :: < Test > :: insert ( old_coldkey, stake) ;
276276
277277 let mut weight = Weight :: zero ( ) ;
278278 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -286,8 +286,8 @@ fn test_swap_idempotency() {
286286 & mut weight
287287 ) ) ;
288288
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) ;
291291 } ) ;
292292}
293293
@@ -299,7 +299,7 @@ fn test_swap_with_max_values() {
299299 let new_coldkey = U256 :: from ( 2 ) ;
300300 let max_stake = u64:: MAX ;
301301
302- TotalColdkeyStake :: < Test > :: insert ( & old_coldkey, max_stake) ;
302+ TotalColdkeyStake :: < Test > :: insert ( old_coldkey, max_stake) ;
303303
304304 let mut weight = Weight :: zero ( ) ;
305305 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -308,8 +308,8 @@ fn test_swap_with_max_values() {
308308 & mut weight
309309 ) ) ;
310310
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) ;
313313 } ) ;
314314}
315315
@@ -321,7 +321,7 @@ fn test_swap_with_non_existent_new_coldkey() {
321321 let new_coldkey = U256 :: from ( 2 ) ;
322322 let stake = 100 ;
323323
324- TotalColdkeyStake :: < Test > :: insert ( & old_coldkey, stake) ;
324+ TotalColdkeyStake :: < Test > :: insert ( old_coldkey, stake) ;
325325
326326 let mut weight = Weight :: zero ( ) ;
327327 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -330,8 +330,8 @@ fn test_swap_with_non_existent_new_coldkey() {
330330 & mut weight
331331 ) ) ;
332332
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) ;
335335 } ) ;
336336}
337337
@@ -343,8 +343,8 @@ fn test_swap_with_overflow_in_stake_addition() {
343343 let new_coldkey = U256 :: from ( 2 ) ;
344344 let max_stake = u64:: MAX ;
345345
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 ) ;
348348
349349 let mut weight = Weight :: zero ( ) ;
350350 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -353,8 +353,8 @@ fn test_swap_with_overflow_in_stake_addition() {
353353 & mut weight
354354 ) ) ;
355355
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) ;
358358 } ) ;
359359}
360360
@@ -367,7 +367,7 @@ fn test_swap_with_max_hotkeys() {
367367 let max_hotkeys = 1000 ;
368368 let hotkeys: Vec < U256 > = ( 0 ..max_hotkeys) . map ( U256 :: from) . collect ( ) ;
369369
370- OwnedHotkeys :: < Test > :: insert ( & old_coldkey, hotkeys. clone ( ) ) ;
370+ OwnedHotkeys :: < Test > :: insert ( old_coldkey, hotkeys. clone ( ) ) ;
371371
372372 let mut weight = Weight :: zero ( ) ;
373373 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -376,8 +376,8 @@ fn test_swap_with_max_hotkeys() {
376376 & mut weight
377377 ) ) ;
378378
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) ;
381381 } ) ;
382382}
383383
@@ -391,10 +391,10 @@ fn test_swap_effect_on_delegated_stake() {
391391 let hotkey = U256 :: from ( 4 ) ;
392392 let stake = 100 ;
393393
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) ;
398398
399399 let mut weight = Weight :: zero ( ) ;
400400 assert_ok ! ( SubtensorModule :: perform_swap_coldkey(
@@ -403,9 +403,9 @@ fn test_swap_effect_on_delegated_stake() {
403403 & mut weight
404404 ) ) ;
405405
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 ) ;
409409 } ) ;
410410}
411411
@@ -420,8 +420,8 @@ fn test_swap_concurrent_modifications() {
420420 let initial_stake = 100 ;
421421 let additional_stake = 50 ;
422422
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) ;
425425
426426 // Simulate concurrent stake addition
427427 add_network ( netuid, 1 , 1 ) ;
@@ -441,10 +441,10 @@ fn test_swap_concurrent_modifications() {
441441 ) ) ;
442442
443443 assert_eq ! (
444- Stake :: <Test >:: get( & hotkey, & new_coldkey) ,
444+ Stake :: <Test >:: get( hotkey, new_coldkey) ,
445445 initial_stake + additional_stake - 1
446446 ) ;
447- assert ! ( !Stake :: <Test >:: contains_key( & hotkey, & old_coldkey) ) ;
447+ assert ! ( !Stake :: <Test >:: contains_key( hotkey, old_coldkey) ) ;
448448 } ) ;
449449}
450450
0 commit comments