2018: add tests

2018/aoc01/src/main.rs

@@ -33,13 +33,129 @@ fn part2(input: &str) -> Result<i32> {
     let mut freq = 0;
     loop {
         for line in input.lines() {
-            freq += line.parse::<i32>()?;
             if freqs.contains(&freq) {
                 println!("{}", freq);
                 return Ok(freq);
             } else {
                 freqs.insert(freq);
             }
+            freq += line.parse::<i32>()?;
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn part1_provided1() {
+        let input = "+1
+-2
++3
++1
+";
+
+        assert_eq!(part1(input).unwrap(), 3);
+    }
+
+    #[test]
+    fn part1_provided2() {
+        let input = "+1
++1
++1
+";
+
+        assert_eq!(part1(input).unwrap(), 3);
+    }
+
+    #[test]
+    fn part1_provided3() {
+        let input = "+1
++1
+-2
+";
+
+        assert_eq!(part1(input).unwrap(), 0);
+    }
+
+    #[test]
+    fn part1_provided4() {
+        let input = "-1
+-2
+-3
+";
+
+        assert_eq!(part1(input).unwrap(), -6);
+    }
+
+    #[test]
+    fn part1_real() {
+        let input = include_str!("../input/input.txt");
+
+        assert_eq!(part1(input).unwrap(), 427);
+    }
+
+    #[test]
+    fn part2_provided1() {
+        let input = "+1
+-2
++3
++1
+";
+
+        assert_eq!(part2(input).unwrap(), 2);
+    }
+
+    #[test]
+    fn part2_provided2() {
+        let input = "+1
+-1
+";
+
+        assert_eq!(part2(input).unwrap(), 0);
+    }
+
+    #[test]
+    fn part2_provided3() {
+        let input = "+3
++3
++4
+-2
+-4
+";
+
+        assert_eq!(part2(input).unwrap(), 10);
+    }
+
+    #[test]
+    fn part2_provided4() {
+        let input = "-6
++3
++8
++5
+-6
+";
+
+        assert_eq!(part2(input).unwrap(), 5);
+    }
+
+    #[test]
+    fn part2_provided5() {
+        let input = "+7
++7
+-2
+-7
+-4
+";
+
+        assert_eq!(part2(input).unwrap(), 14);
+    }
+
+    #[test]
+    fn part2_real() {
+        let input = include_str!("../input/input.txt");
+
+        assert_eq!(part2(input).unwrap(), 341);
+    }
+}

2018/aoc02/src/main.rs

@@ -40,3 +40,29 @@ fn part1(input: &str) -> Result<u32> {
 
     Ok(twice * thrice)
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn part1_provided() {
+        let input = "abcdef
+bababc
+abbcde
+abcccd
+aabcdd
+abcdee
+ababab
+";
+
+        assert_eq!(part1(input).unwrap(), 12);
+    }
+
+    #[test]
+    fn part1_real() {
+        let input = include_str!("../input/input.txt");
+
+        assert_eq!(part1(input).unwrap(), 5750);
+    }
+}