From 5c62892fbd2393036d3380c874eef59384ea3641 Mon Sep 17 00:00:00 2001 From: Curtis McEnroe Date: Sat, 3 Dec 2016 23:20:01 -0500 Subject: Add test to day 2 --- src/bin/day02.rs | 22 ++++++++++++++++------ 1 file changed, 16 insertions(+), 6 deletions(-) (limited to 'src/bin') diff --git a/src/bin/day02.rs b/src/bin/day02.rs index c5a7f11..21d101b 100644 --- a/src/bin/day02.rs +++ b/src/bin/day02.rs @@ -1,6 +1,6 @@ use std::io::{self, Read}; -#[derive(Debug, Clone, Copy)] +#[derive(Debug, Clone, Copy, PartialEq, Eq)] enum Keypad { K1, K2, K3, K4, K5, K6, @@ -47,10 +47,7 @@ impl Keypad { } } -fn main() { - let mut input = String::new(); - io::stdin().read_to_string(&mut input).unwrap(); - +fn solve(input: &str) -> Vec { let mut code = Vec::new(); let mut key = Keypad::K5; @@ -67,5 +64,18 @@ fn main() { code.push(key); } - println!("Part 1: {:?}", code); + code +} + +fn main() { + let mut input = String::new(); + io::stdin().read_to_string(&mut input).unwrap(); + + println!("Part 1: {:?}", solve(&input)); +} + +#[test] +fn part1() { + use Keypad::*; + assert_eq!(vec![K1, K9, K8, K5], solve("ULL\nRRDDD\nLURDL\nUUUUD")); } -- cgit 1.4.1