33use std:: u16;
44
55use frame_support:: traits:: Currency ;
6- use substrate_fixed:: types:: extra:: True ;
76
87use crate :: mock:: * ;
98use frame_support:: dispatch:: { DispatchClass , DispatchInfo , GetDispatchInfo , Pays } ;
@@ -551,6 +550,9 @@ fn test_burn_registration_pruning_scenarios() {
551550 let max_allowed_uids = 6 ;
552551 let immunity_period = 5000 ;
553552
553+ const IS_IMMUNE : bool = true ;
554+ const NOT_IMMUNE : bool = false ;
555+
554556 // Initial setup
555557 SubtensorModule :: set_burn ( netuid, burn_cost) ;
556558 SubtensorModule :: set_max_allowed_uids ( netuid, max_allowed_uids) ;
@@ -575,9 +577,9 @@ fn test_burn_registration_pruning_scenarios() {
575577 // Note: pruning score is set to u16::MAX after getting neuron to prune
576578
577579 // 1. Test if all immune neurons
578- assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 0 ) , true ) ;
579- assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 1 ) , true ) ;
580- assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 2 ) , true ) ;
580+ assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 0 ) , IS_IMMUNE ) ;
581+ assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 1 ) , IS_IMMUNE ) ;
582+ assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 2 ) , IS_IMMUNE ) ;
581583
582584 SubtensorModule :: set_pruning_score_for_uid ( netuid, 0 , 100 ) ;
583585 SubtensorModule :: set_pruning_score_for_uid ( netuid, 1 , 75 ) ;
@@ -597,9 +599,9 @@ fn test_burn_registration_pruning_scenarios() {
597599 step_block ( immunity_period) ;
598600
599601 // ensure all neurons are non-immune
600- assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 0 ) , false ) ;
601- assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 1 ) , false ) ;
602- assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 2 ) , false ) ;
602+ assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 0 ) , NOT_IMMUNE ) ;
603+ assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 1 ) , NOT_IMMUNE ) ;
604+ assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 2 ) , NOT_IMMUNE ) ;
603605
604606 SubtensorModule :: set_pruning_score_for_uid ( netuid, 0 , 100 ) ;
605607 SubtensorModule :: set_pruning_score_for_uid ( netuid, 1 , 50 ) ;
@@ -627,9 +629,9 @@ fn test_burn_registration_pruning_scenarios() {
627629 }
628630
629631 // Ensure all new neurons are immune
630- assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 3 ) , true ) ;
631- assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 4 ) , true ) ;
632- assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 5 ) , true ) ;
632+ assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 3 ) , IS_IMMUNE ) ;
633+ assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 4 ) , IS_IMMUNE ) ;
634+ assert_eq ! ( SubtensorModule :: get_neuron_is_immune( netuid, 5 ) , IS_IMMUNE ) ;
633635
634636 // Set pruning scores for all neurons
635637 SubtensorModule :: set_pruning_score_for_uid ( netuid, 0 , 75 ) ; // non-immune
0 commit comments