From 2ab1f41efeaba57e738fa1c23f60deda99cc9730 Mon Sep 17 00:00:00 2001
From: Adam Jeniski <ajeniski4@gmail.com>
Date: Fri, 7 Jan 2022 02:46:33 -0500
Subject: [PATCH] fix tests

---
 src/day1.rs | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/day1.rs b/src/day1.rs
index 6f9b66f..e25f57b 100644
--- a/src/day1.rs
+++ b/src/day1.rs
@@ -11,12 +11,29 @@ mod part1 {
     #[test]
     fn run_sample() {
         let input = include_str!("../inputs/1_test.txt");
-        assert_eq!(part_1_solution(input), 0);
+        assert_eq!(part_1_solution(input), 7);
     }
 
     #[test]
     fn run() {
         let input = include_str!("../inputs/1.txt");
-        assert_eq!(part_1_solution(input), 0);
+        assert_eq!(part_1_solution(input), 1139);
+    }
+}
+
+#[cfg(test)]
+mod part2 {
+    use super::*;
+
+    #[test]
+    fn run_sample() {
+        let input = include_str!("../inputs/1_test.txt");
+        assert_eq!(part_2_solution(input), 5);
+    }
+
+    #[test]
+    fn run() {
+        let input = include_str!("../inputs/1.txt");
+        assert_eq!(part_2_solution(input), 1103);
     }
 }