Refactoring

src/block_step.rs

@@ -20,6 +20,9 @@ use substrate_fixed::types::I110F18;
 use substrate_fixed::types::I64F64;
 use substrate_fixed::types::I96F32;
 
+#[cfg(test)]
+use crate::state::{TOTAL_COLDKEY_STAKE, TOTAL_STAKE};
+
 /// Executes the necessary operations for each block.
 pub fn block_step(deps: DepsMut, env: Env, caller: Option<Addr>) -> Result<Response, ContractError> {
     if caller.is_some() {

src/epoch.rs

@@ -23,6 +23,8 @@ use crate::utils::{
 
 #[cfg(test)]
 use crate::uids::get_stake_for_uid_and_subnetwork;
+#[cfg(test)]
+use crate::utils::get_rho;
 
 // Calculates reward consensus values, then updates rank, trust, consensus, incentive, dividend, pruning_score, emission and bonds, and
 // returns the emissions for uids/hotkeys in a given `netuid`.

src/uids.rs

@@ -9,6 +9,9 @@ use crate::state::{
 use crate::utils::set_active_for_uid;
 use crate::ContractError;
 
+#[cfg(test)]
+use crate::state::TOTAL_HOTKEY_STAKE;
+
 pub fn get_subnetwork_n(store: &dyn Storage, netuid: u16) -> u16 {
     SUBNETWORK_N.load(store, netuid.clone()).unwrap()
 }

src/utils.rs

@@ -24,6 +24,11 @@ use crate::state::{
 };
 use crate::uids::get_subnetwork_n;
 
+#[cfg(test)]
+use crate::state::{LAST_ADJUSTMENT_BLOCK, POW_REGISTRATIONS_THIS_INTERVAL, BURN_REGISTRATIONS_THIS_INTERVAL,
+    LAST_MECHANISM_STEP_BLOCK, PENDING_EMISSION,
+};
+
 pub fn ensure_subnet_owner_or_root(
     store: &dyn Storage,
     coldkey: &Addr,