1 files changed, 25 insertions, 9 deletions

diff --git a/aoc/src/lib.rs b/aoc/src/lib.rs
index c4318e1..dcca651 100644
--- a/aoc/src/lib.rs
+++ b/aoc/src/lib.rs
@@ -2,6 +2,7 @@ use clap::Parser;
 use std::fs::File;
 use std::io;
 use std::io::{BufRead, BufReader, Read};
+use std::panic::catch_unwind;
 use std::time::Instant;
 
 #[derive(Parser, Debug)]
@@ -16,6 +17,14 @@ struct Args {
 
 pub trait Day {
     fn example_input(&self) -> &'static str;
+    fn example_input_part_1(&self) -> &'static str {
+        self.example_input()
+    }
+
+    fn example_input_part_2(&self) -> &'static str {
+        self.example_input()
+    }
+
     fn example_result_part_1(&self) -> &'static str;
     fn example_result_part_2(&self) -> &'static str;
 
@@ -61,19 +70,26 @@ pub fn main(day: &dyn Day) {
     println!("Time to run: {}s", now.elapsed().as_secs_f64());
 }
 
-pub fn test_day<T: Day>(day: &T) {
+pub fn test_day<T: Day + std::panic::RefUnwindSafe>(day: &T) {
     use std::io::Cursor;
     use std::io::{BufReader, Read};
 
-    let example_input = day.example_input();
-
     // Test Part 1
-    let input = BufReader::new(Box::new(Cursor::new(example_input)) as Box<dyn Read>);
-    let output = day.part_1(input);
-    assert_eq!(output.trim(), day.example_result_part_1(), "Part 1 failed");
+    let output = catch_unwind(|| {
+        let input = BufReader::new(Box::new(Cursor::new(day.example_input_part_1())) as Box<dyn Read>);
+        day.part_1(input)
+    });
+    if output.is_ok() {
+        assert_eq!(output.unwrap().trim(), day.example_result_part_1(), "Part 1 failed");
+    }
 
     // Test Part 2
-    let input = BufReader::new(Box::new(Cursor::new(example_input)) as Box<dyn Read>);
-    let output = day.part_2(input);
-    assert_eq!(output.trim(), day.example_result_part_2(), "Part 2 failed");
+    let output = catch_unwind(|| {
+        let input = BufReader::new(Box::new(Cursor::new(day.example_input_part_2())) as Box<dyn Read>);
+        day.part_2(input)
+    });
+
+    if output.is_ok() {
+        assert_eq!(output.unwrap().trim(), day.example_result_part_2(), "Part 2 failed");
+    }
 }
\ No newline at end of file