some test refactoring

Author: kirjavascript

tests/src/main.rs

@@ -3,6 +3,7 @@ mod input;
 mod labels;
 mod pushdown;
 mod rng;
+mod score;
 mod sps;
 mod util;
 mod video;
@@ -63,18 +64,17 @@ fn main() {
 
     // other stuff
 
-    let rng_seed = labels::get("rng_seed") as usize;
-    let main_loop = labels::get("mainLoop");
-    let practise_type = labels::get("practiseType") as usize;
-    let game_mode = labels::get("gameMode") as usize;
-    let level_number = labels::get("levelNumber") as usize;
-    let b_modifier = labels::get("typeBModifier") as usize;
-    let mode_typeb = labels::get("MODE_TYPEB") as u8;
-
     if options.foo {
         let mut emu = util::emulator(None);
         let mut view = video::Video::new();
 
+        let rng_seed = labels::get("rng_seed") as usize;
+        let main_loop = labels::get("mainLoop");
+        let practise_type = labels::get("practiseType") as usize;
+        let game_mode = labels::get("gameMode") as usize;
+        let level_number = labels::get("levelNumber") as usize;
+        let b_modifier = labels::get("typeBModifier") as usize;
+        let mode_typeb = labels::get("MODE_TYPEB") as u8;
 
         rng::seeds().iter().for_each(|seed| {
 

tests/src/pushdown.rs

@@ -1,12 +1,12 @@
-use crate::{labels, util};
+use crate::{labels, util, score};
 
 pub fn test() {
     let mut emu = util::emulator(None);
 
     for pushdown in 2..15 {
         [0..1000, 24500..25500].into_iter().for_each(|range| {
             for score in range {
-                util::set_score(&mut emu, score);
+                score::set(&mut emu, score);
 
                 emu.registers.pc = labels::get("addPushDownPoints");
                 emu.memory.iram_raw[labels::get("holdDownPoints") as usize] = pushdown;
@@ -15,7 +15,7 @@ pub fn test() {
 
                 let reference = pushdown_impl(pushdown, score as u16) as u32;
 
-                assert_eq!(reference, util::get_score(&mut emu) - score);
+                assert_eq!(reference, score::get(&mut emu) - score);
             }
         });
     }

tests/src/rng.rs

@@ -1,6 +1,16 @@
 use crate::{util, labels};
 use std::collections::HashSet;
 
+pub fn test() {
+    assert_eq!(
+        seeds(),
+        include_str!("./rng_seeds.txt")
+            .split(',')
+            .map(|s| s.trim().parse::<u16>().expect(s))
+            .collect::<HashSet<_>>()
+    );
+}
+
 pub fn seeds() -> HashSet<u16> {
     let mut emu = util::emulator(None);
     let rng_seed = labels::get("rng_seed");
@@ -33,13 +43,3 @@ pub fn seeds() -> HashSet<u16> {
 
     seeds
 }
-
-pub fn test() {
-    assert_eq!(
-        seeds(),
-        include_str!("./rng_seeds.txt")
-            .split(',')
-            .map(|s| s.trim().parse::<u16>().expect(s))
-            .collect::<HashSet<_>>()
-    );
-}

tests/src/score.rs

@@ -0,0 +1,25 @@
+use rusticnes_core::nes::NesState;
+use crate::labels;
+
+pub fn set(emu: &mut NesState, score: u32) {
+    let score_addr = labels::get("score");
+    let binscore_addr = labels::get("binScore");
+    let bcd_str = format!("{:06}", score);
+    let bcd_a = i64::from_str_radix(&bcd_str[0..2], 16).unwrap();
+    let bcd_b = i64::from_str_radix(&bcd_str[2..4], 16).unwrap();
+    let bcd_c = i64::from_str_radix(&bcd_str[4..6], 16).unwrap();
+
+    emu.memory.iram_raw[(score_addr + 2) as usize] = bcd_a as u8;
+    emu.memory.iram_raw[(score_addr + 1) as usize] = bcd_b as u8;
+    emu.memory.iram_raw[score_addr as usize] = bcd_c as u8;
+    emu.memory.iram_raw[binscore_addr as usize] = score as u8;
+    emu.memory.iram_raw[(binscore_addr + 1) as usize] = (score >> 8) as u8;
+    emu.memory.iram_raw[(binscore_addr + 2) as usize] = (score >> 16) as u8;
+}
+
+pub fn get(emu: &mut NesState) -> u32 {
+    let binscore_addr = labels::get("binScore");
+    emu.memory.iram_raw[binscore_addr as usize] as u32
+        + ((emu.memory.iram_raw[(binscore_addr + 1) as usize] as u32) << 8)
+        + ((emu.memory.iram_raw[(binscore_addr + 2) as usize] as u32) << 16)
+}

tests/src/util.rs

@@ -42,26 +42,3 @@ pub fn print_step(emu: &mut NesState) {
 
     println!("{}", opcode_info::disassemble_instruction(emu.cpu.opcode, 0, 0).0);
 }
-
-pub fn set_score(emu: &mut NesState, score: u32) {
-    let score_addr = labels::get("score");
-    let binscore_addr = labels::get("binScore");
-    let bcd_str = format!("{:06}", score);
-    let bcd_a = i64::from_str_radix(&bcd_str[0..2], 16).unwrap();
-    let bcd_b = i64::from_str_radix(&bcd_str[2..4], 16).unwrap();
-    let bcd_c = i64::from_str_radix(&bcd_str[4..6], 16).unwrap();
-
-    emu.memory.iram_raw[(score_addr + 2) as usize] = bcd_a as u8;
-    emu.memory.iram_raw[(score_addr + 1) as usize] = bcd_b as u8;
-    emu.memory.iram_raw[score_addr as usize] = bcd_c as u8;
-    emu.memory.iram_raw[binscore_addr as usize] = score as u8;
-    emu.memory.iram_raw[(binscore_addr + 1) as usize] = (score >> 8) as u8;
-    emu.memory.iram_raw[(binscore_addr + 2) as usize] = (score >> 16) as u8;
-}
-
-pub fn get_score(emu: &mut NesState) -> u32 {
-    let binscore_addr = labels::get("binScore");
-    emu.memory.iram_raw[binscore_addr as usize] as u32
-        + ((emu.memory.iram_raw[(binscore_addr + 1) as usize] as u32) << 8)
-        + ((emu.memory.iram_raw[(binscore_addr + 2) as usize] as u32) << 16)
-}