Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

refac(pallets/subspace): refactor test_burn #26

Merged
merged 1 commit into from
Mar 8, 2024
Merged

refac(pallets/subspace): refactor test_burn #26

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
81 changes: 14 additions & 67 deletions pallets/subspace/tests/burn.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@ use frame_support::assert_ok;

use mock::*;
use sp_core::U256;
use sp_std::vec;

#[test]
fn test_burn() {
Expand All @@ -14,91 +13,40 @@ fn test_burn() {
let tempo = 100;

let keys: Vec<U256> = (0..n).map(U256::from).collect();
let stakes: Vec<u64> = (0..n).map(|_x| initial_stake * 1_000_000_000).collect();

let stakes: Vec<u64> = std::iter::repeat(initial_stake * 1_000_000_000).take(n).collect();
let voter_key = keys[1];
let mut subnet_params = SubspaceModule::subnet_params(netuid);
subnet_params.tempo = tempo;

SubspaceModule::set_subnet_params(netuid, subnet_params);

for i in 0..n {
assert_ok!(register_module(netuid, keys[i], stakes[i]));
let stake_from_vector = SubspaceModule::get_stake_to_vector(netuid, &keys[i]);
println!("{:?}", stake_from_vector);
for (key, stake) in keys.iter().zip(stakes.iter()) {
assert_ok!(register_module(netuid, *key, *stake));
}

for burn_rate in [0, 10, 20, 30, 40, 50, 60, 70, 80, 90].iter() {
for burn_rate in (0..9).map(|n| n * 10) {
let mut params = SubspaceModule::global_params();
params.burn_rate = *burn_rate;
SubspaceModule::set_global_params(params);
params.burn_rate = burn_rate;

let voter_key = keys[1];
SubspaceModule::set_global_params(params);

// vote to avoid key[0] as we want to see the key[0] burn
let mut votes: Vec<u16> = vec![];
let mut uids: Vec<u16> = vec![];
for i in 0..n {
votes.push(1);
uids.push(i as u16);
}
println!(
"{:?}",
SubspaceModule::get_stake_for_key(netuid, &voter_key)
);
let votes: Vec<u16> = std::iter::repeat(1).take(n).collect();
let uids: Vec<u16> = (0..n).map(|x| x as u16).collect();

assert_ok!(SubspaceModule::set_weights(
get_origin(voter_key),
netuid,
uids,
votes
));
println!(
"burn : {:?}",
SubspaceModule::get_burn_emission_per_epoch(netuid)
);

let _total_stake = SubspaceModule::get_total_subnet_stake(netuid);

let dividends = SubspaceModule::get_dividends(netuid);
let incentives = SubspaceModule::get_incentives(netuid);
let emissions = SubspaceModule::get_emissions(netuid);

println!(
"dividends: {:?} incentives: {:?} emissions: {:?}",
dividends, incentives, emissions
);

let burn_per_epoch = SubspaceModule::get_burn_per_epoch(netuid);

println!("burn_per_epoch: {:?}", burn_per_epoch);

let stake_vector_before = SubspaceModule::get_stakes(netuid);
let total_stake_before = SubspaceModule::get_total_subnet_stake(netuid);
step_epoch(netuid);
let stake_vector_after = SubspaceModule::get_stakes(netuid);
let total_stake_after = SubspaceModule::get_total_subnet_stake(netuid);

println!(
"stake_vector_before: {:?} stake_vector_after: {:?}",
stake_vector_before, stake_vector_after
);
println!(
"total_stake: {:?} total_stake_after: {:?}",
total_stake_before, total_stake_after
);
let total_stake_after = SubspaceModule::get_total_subnet_stake(netuid);
let subnet_params = SubspaceModule::subnet_params(netuid);

let burn_per_epoch = SubspaceModule::get_burn_emission_per_epoch(netuid);
let dividends = SubspaceModule::get_dividends(netuid);
let incentives = SubspaceModule::get_incentives(netuid);
let emissions = SubspaceModule::get_emissions(netuid);

println!("burn_per_epoch: {:?}", burn_per_epoch);

println!(
"dividends: {:?} incentives: {:?} emissions: {:?}",
dividends, incentives, emissions
);

let _calculated_subnet_emission = emissions.iter().sum::<u64>();
let expected_subnet_emission: u64 =
((SubspaceModule::get_subnet_emission(netuid) as f64
* (subnet_params.tempo as f64))
Expand All @@ -109,12 +57,11 @@ fn test_burn() {

let expected_subnet_emission =
total_stake_before.saturating_add(expected_subnet_emission);

assert!(
total_stake_after.saturating_sub(delta) < expected_subnet_emission
&& total_stake_after + delta > expected_subnet_emission,
"total_stake_after: {:?} expected_subnet_emission: {:?}",
total_stake_after,
expected_subnet_emission
"total_stake_after: {total_stake_after:?} expected_subnet_emission: {expected_subnet_emission:?}",
);
}
});
Expand Down
Loading