test sps

Author: kirjavascript

tests/src/block.rs

@@ -17,3 +17,18 @@ impl From<u8> for Block {
         }
     }
 }
+
+impl From<char> for Block {
+    fn from(value: char) -> Self {
+        match value {
+            'T' => Block::T,
+            'J' => Block::J,
+            'Z' => Block::Z,
+            'O' => Block::O,
+            'S' => Block::S,
+            'L' => Block::L,
+            'I' => Block::I,
+            _ => Block::Unknown(value as u8),
+        }
+    }
+}

tests/src/main.rs

@@ -24,11 +24,15 @@ struct TestOptions {
 fn main() {
     let options = TestOptions::parse_args_default_or_exit();
 
+    // run SPS tests
     if options.test {
+        sps::test();
+        println!("sps works!");
         pushdown::test();
         println!("pushdown works!");
     }
 
+    // print SPS sequences
     if options.sps_qty > 0 {
         let mut blocks = sps::SPS::new();
 

tests/src/sps.rs

@@ -1,5 +1,5 @@
+use crate::{block, labels, util};
 use rusticnes_core::nes::NesState;
-use crate::{labels, util, block};
 
 pub struct SPS {
     emu: NesState,
@@ -29,3 +29,27 @@ impl SPS {
         self.emu.memory.iram_raw[labels::get("spawnID") as usize].into()
     }
 }
+
+pub fn test() {
+    let mut blocks = SPS::new();
+
+    blocks.set_input((0x10, 0x10, 0x10));
+    "ZJOTLTZJLZJSZISIJOLITJSILZJILITSISZOITIZSZJLLTIOZJZSZISIJZTIZJTSOJSJISJOOTSJTOTZSZTZSLTZTOTSIZJZIJIL".chars().for_each(|block| {
+        assert_eq!(blocks.next(), block.into());
+    });
+
+    blocks.set_input((0x12, 0x34, 0x56));
+    "ZTZIJIJOZTSOSZJZOSLIOIJIJSTZSTTJISSTOIZJITJOZJITSOSZSJLTISJOITTLSLJTZTZOZSLJTJZSLTSOTLOJLSJSJTJILOJS".chars().for_each(|block| {
+        assert_eq!(blocks.next(), block.into());
+    });
+
+    blocks.set_input((0x87, 0xAB, 0x12));
+    "OZIJSOTZSJTSTJZLOLJOJISOZOIOZJITILSSJZLOIJSTITLSOJILTSOOLZOOIJOZLTLSISIJIJTOLSIJILSLOLJLTOSOSLOIZSIS".chars().for_each(|block| {
+        assert_eq!(blocks.next(), block.into());
+    });
+
+    blocks.set_input((0x13, 0x37, 0x02));
+    "OJSTZSIOLSIJTSZILJZJJLZLISISJTLZTSZTJOJOSJSZLITJOIOTITILTOSTJSZTSOOIOJSIITLJOZSIOJOTZTLJLIOJLITSSLSLIJIIOLOISLZJLIJJTIZIJOJISLTIJTOTZIOILSTTLTZIZJSOLOZOZOLOTZTZOTZOSIOTJJTSIZSOLTOLIZSOZOTZISLJTSZLOISO".chars().for_each(|block| {
+        assert_eq!(blocks.next(), block.into());
+    });
+}