diff --git a/aoc2021/input/day01_provided.txt b/aoc2021/input/day01_provided.txt
new file mode 100644
index 0000000..167e291
--- /dev/null
+++ b/aoc2021/input/day01_provided.txt
@@ -0,0 +1,10 @@
+199
+200
+208
+210
+200
+207
+240
+269
+260
+263
diff --git a/aoc2021/src/day01.rs b/aoc2021/src/day01.rs
index ba6765e..26f40b3 100644
--- a/aoc2021/src/day01.rs
+++ b/aoc2021/src/day01.rs
@@ -43,3 +43,30 @@ fn part2(input: &str) -> Result<usize> {
 
     Ok(increases)
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    const PROVIDED: &str = include_str!("../input/day01_provided.txt");
+
+    #[test]
+    fn part1_provided() {
+        assert_eq!(part1(PROVIDED).unwrap(), 7);
+    }
+
+    #[test]
+    fn part1_real() {
+        assert_eq!(part1(INPUT).unwrap(), 1502);
+    }
+
+    #[test]
+    fn part2_provided() {
+        assert_eq!(part2(PROVIDED).unwrap(), 5);
+    }
+
+    #[test]
+    fn part2_real() {
+        assert_eq!(part2(INPUT).unwrap(), 1538);
+    }
+}