Update tests with basic init

src/test_roms/mod.rs

@@ -2,12 +2,15 @@ use crate::{NES, hex_view::Memory};
 
 macro_rules! rom_test {
     ($name:ident, $rom:literal, |$nes:ident| $eval:expr) => {
+        rom_test!($name, $rom, timeout = 10000000, |$nes| $eval);
+    };
+    ($name:ident, $rom:literal, timeout = $timeout:expr, |$nes:ident| $eval:expr) => {
         #[test]
         fn $name() {
             let rom_file = concat!(env!("ROM_DIR"), "/", $rom);
             println!("{}: {}", stringify!($name), rom_file);
             let mut $nes = NES::load_nes_file(rom_file).expect("Failed to create nes object");
-            $nes.reset_and_run_with_timeout(1000000);
+            $nes.reset_and_run_with_timeout($timeout);
             println!("Final: {:?}", $nes);
             $eval
         }
@@ -39,3 +42,23 @@ rom_test!(read_write, "read_write.nes", |nes| {
     assert_eq!(nes.cpu_mem().peek(0x0001).unwrap(), 0xAA);
     assert_eq!(nes.cpu_mem().peek(0x0002).unwrap(), 0xAA);
 });
+
+rom_test!(basic_init_0, "basic_init_0.nes", |nes| {
+    assert_eq!(nes.last_instruction, "0x8017 HLT :2 []");
+    assert_eq!(nes.cycle, 41);
+    assert_eq!(nes.cpu.pc, 0x8018);
+    assert_eq!(nes.cpu.sp, 0xFF);
+    assert_eq!(nes.cpu.a, 0x00);
+    assert_eq!(nes.cpu.x, 0x00);
+    assert_eq!(nes.cpu.y, 0x00);
+});
+
+rom_test!(basic_init_1, "basic_init_1.nes", |nes| {
+    assert_eq!(nes.last_instruction, "0x8017 HLT :2 []");
+    assert_eq!(nes.cycle, 41);
+    assert_eq!(nes.cpu.pc, 0x8018);
+    assert_eq!(nes.cpu.sp, 0xFF);
+    assert_eq!(nes.cpu.a, 0x00);
+    assert_eq!(nes.cpu.x, 0x00);
+    assert_eq!(nes.cpu.y, 0x00);
+});