Add stake weights tests

Author: ales-otf

pallets/subtensor/tests/staking.rs

@@ -2494,3 +2494,119 @@ fn test_anneal_global_weight() {
         );
     });
 }
+
+// https://github.com/opentensor/subtensor/issues/925
+// cargo test -p pallet-subtensor --test staking -- test_stake_weight_should_not_be_affected_by_zero_stakes --exact --nocapture
+#[test]
+fn test_stake_weight_should_not_be_affected_by_zero_stakes() {
+    new_test_ext(1).execute_with(|| {
+        let registrar = AccountId::from(42); // SN Owner
+        let netuid: u16 = 1;
+        Balances::force_set_balance(RuntimeOrigin::root(), registrar, 10_000_000_000_000).unwrap();
+
+        // Add root network
+        add_network(0, 0, 0);
+
+        // our test subnet with id 1
+        SubtensorModule::register_network_with_identity(
+            RuntimeOrigin::signed(registrar),
+            registrar,
+            1,
+            None,
+        )
+        .unwrap();
+
+        let neuron_1 = AccountId::from(123);
+        SubtensorModule::add_balance_to_coldkey_account(&neuron_1, 10_000_000_000_000);
+
+        SubtensorModule::burned_register(RuntimeOrigin::signed(neuron_1), netuid, neuron_1)
+            .unwrap();
+        SubtensorModule::add_stake(
+            RuntimeOrigin::signed(neuron_1),
+            neuron_1,
+            netuid,
+            50_000_000,
+        )
+        .unwrap();
+
+        let stake_weights_1 = SubtensorModule::get_stake_weights_for_network(netuid);
+
+        // Check neuron 1's alpha
+        assert!(
+            stake_weights_1.0[1] > 0,
+            "The neuron 1 should get at least some stake"
+        );
+
+        let neuron_2 = AccountId::from(321);
+        SubtensorModule::add_balance_to_coldkey_account(&neuron_2, 10_000_000_000_000);
+
+        SubtensorModule::burned_register(RuntimeOrigin::signed(neuron_2), netuid, neuron_2)
+            .unwrap();
+        SubtensorModule::add_stake(
+            RuntimeOrigin::signed(neuron_2),
+            neuron_2,
+            netuid,
+            50_000_000,
+        )
+        .unwrap();
+
+        let stake_weights_2 = SubtensorModule::get_stake_weights_for_network(netuid);
+
+        // Check neuron 2's alpha
+        assert!(
+            stake_weights_2.0[2] > 0,
+            "The neuron 2 should get at least some stake"
+        );
+    });
+}
+
+// cargo test -p pallet-subtensor --test staking -- test_stake_weight_no_subnet_stake --exact --nocapture
+#[test]
+fn test_stake_weight_no_subnet_stake() {
+    new_test_ext(1).execute_with(|| {
+        let registrar = AccountId::from(42); // SN Owner
+        let netuid: u16 = 1;
+        Balances::force_set_balance(RuntimeOrigin::root(), registrar, 10_000_000_000_000).unwrap();
+
+        // Add root network
+        add_network(0, 0, 0);
+
+        // our test subnet with id 1
+        SubtensorModule::register_network_with_identity(
+            RuntimeOrigin::signed(registrar),
+            registrar,
+            1,
+            None,
+        )
+        .unwrap();
+
+        let neuron_1 = AccountId::from(123);
+        SubtensorModule::add_balance_to_coldkey_account(&neuron_1, 10_000_000_000_000);
+
+        SubtensorModule::burned_register(RuntimeOrigin::signed(neuron_1), netuid, neuron_1)
+            .unwrap();
+
+        let stake_weights_1 = SubtensorModule::get_stake_weights_for_network(netuid);
+
+        // Check neuron 1's alpha
+        assert!(
+            stake_weights_1.0[1] == 0,
+            "The neuron 1's stake should be 0"
+        );
+
+        let neuron_2 = AccountId::from(321);
+        SubtensorModule::add_balance_to_coldkey_account(&neuron_2, 10_000_000_000_000);
+
+        SubtensorModule::burned_register(RuntimeOrigin::signed(neuron_2), netuid, neuron_2)
+            .unwrap();
+
+        let stake_weights_2 = SubtensorModule::get_stake_weights_for_network(netuid);
+
+        // Check neuron 2's alpha
+        assert!(
+            stake_weights_2.0[2] == 0,
+            "The neuron 2's stake should be 0"
+        );
+    });
+}
+