From 176012f5ea81b1e9788b3269899facf34cab6869 Mon Sep 17 00:00:00 2001
From: Curtis McEnroe <programble@gmail.com>
Date: Sat, 3 Dec 2016 23:15:39 -0500
Subject: Add tests to day 1

---
 src/bin/day01.rs | 37 +++++++++++++++++++++++++++----------
 1 file changed, 27 insertions(+), 10 deletions(-)

(limited to 'src')

diff --git a/src/bin/day01.rs b/src/bin/day01.rs
index 26dd306..c596099 100644
--- a/src/bin/day01.rs
+++ b/src/bin/day01.rs
@@ -14,8 +14,8 @@ impl Point {
         Point(-self.1, self.0)
     }
 
-    fn distance(self, other: Self) -> i32 {
-        (self.0 - other.0).abs() + (self.1 - other.1).abs()
+    fn distance(self) -> i32 {
+        self.0.abs() + self.1.abs()
     }
 }
 
@@ -27,10 +27,7 @@ impl Add for Point {
     }
 }
 
-fn main() {
-    let mut input = String::new();
-    io::stdin().read_to_string(&mut input).unwrap();
-
+fn solve(input: &str) -> (i32, Option<i32>) {
     let mut position = Point(0, 0);
     let mut direction = Point(0, -1);
     let mut visited = HashSet::new();
@@ -48,14 +45,34 @@ fn main() {
         let count: i32 = count.parse().unwrap();
         for _ in 0..count {
             position = position + direction;
-            if !visited.insert(position) && collision.is_none() {
+            if collision.is_none() && !visited.insert(position) {
                 collision = Some(position);
             }
         }
     }
 
-    println!("Part 1: {}", position.distance(Point(0, 0)));
-    if let Some(point) = collision {
-        println!("Part 2: {}", point.distance(Point(0, 0)));
+    (position.distance(), collision.map(Point::distance))
+}
+
+fn main() {
+    let mut input = String::new();
+    io::stdin().read_to_string(&mut input).unwrap();
+
+    let (part1, part2) = solve(&input);
+    println!("Part 1: {}", part1);
+    if let Some(part2) = part2 {
+        println!("Part 2: {}", part2);
     }
 }
+
+#[test]
+fn part1() {
+    assert_eq!(5, solve("R2, L3").0);
+    assert_eq!(2, solve("R2, R2, R2").0);
+    assert_eq!(12, solve("R5, L5, R5, R3").0);
+}
+
+#[test]
+fn part2() {
+    assert_eq!(Some(4), solve("R8, R4, R4, R8").1);
+}
-- 
cgit 1.4.1

wmsg=1&amp;follow=1'>root</a>/<a href='/src/log/bin?id=a9c4586213c1d79394d76e3ad6fdb31885d26ccd&amp;showmsg=1&amp;follow=1'>bin</a>/<a href='/src/log/bin/man1?id=a9c4586213c1d79394d76e3ad6fdb31885d26ccd&amp;showmsg=1&amp;follow=1'>man1</a>/<a href='/src/log/bin/man1/brot.1?id=a9c4586213c1d79394d76e3ad6fdb31885d26ccd&amp;showmsg=1&amp;follow=1'>brot.1</a> (<a href='/src/log/bin/man1/brot.1?id=a9c4586213c1d79394d76e3ad6fdb31885d26ccd&amp;showmsg=1'>unfollow</a>)</div><div class='content'><table class='list nowrap'><tr class='nohover'><th></th><th class='left'>Commit message (<a href='/src/log/bin/man1/brot.1?id=a9c4586213c1d79394d76e3ad6fdb31885d26ccd&amp;follow=1'>Collapse</a>)</th><th class='left'>Author</th></tr>
<tr class='logheader'><td><span title='2019-07-09 23:35:52 -0400'>2019-07-09</span></td><td class='logsubject'><a href='/src/commit/bin/up.sh?id=539da9d430ffb5afbd1120eaa9987ad7fa8481ce&amp;follow=1'>Add up -c</a></td><td>June McEnroe</td></tr>
<tr class='nohover-highlight'><td/><td colspan='3' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2019-07-09 23:18:51 -0400'>2019-07-09</span></td><td class='logsubject'><a href='/src/commit/bin/shotty.c?id=f75cb431095abfa56e412b63ec17262e71b97039&amp;follow=1'>Add options for default colors to shotty</a></td><td>June McEnroe</td></tr>
<tr class='nohover-highlight'><td/><td colspan='3' class='logmsg'>
</td></tr>
<tr class='logheader'><td><span title='2019-07-08 22:48:04 -0400'>2019-07-08</span></td><td class='logsubject'><a href='/src/commit/bin/shotty.c?id=5730b2edac38861c9aa0f768a692864417500958&amp;follow=1'>Use char literals consistently</a></td><td>June McEnroe