[rust][7.0] Make the test circuit work!
diff --git a/rust/src/day_07.rs b/rust/src/day_07.rs
index 59b5e75..6fe2b28 100644
--- a/rust/src/day_07.rs
+++ b/rust/src/day_07.rs
@@ -2,7 +2,6 @@ use std::collections::HashMap;
use std::ops::Index;
#[test]
-#[ignore]
fn test_solve() {
let input = "123 -> x
456 -> y
@@ -13,12 +12,19 @@ y RSHIFT 2 -> g
NOT x -> h
NOT y -> i";
+ assert_eq!(72, solve(input, "d"));
+ assert_eq!(507, solve(input, "e"));
+ assert_eq!(492, solve(input, "f"));
+ assert_eq!(114, solve(input, "g"));
+ assert_eq!(65412, solve(input, "h"));
+ assert_eq!(65079, solve(input, "i"));
assert_eq!(123, solve(input, "x"));
+ assert_eq!(456, solve(input, "y"));
}
-pub fn solve(input: &str, wire: &str) -> u16 {
- let circuit = Circuit::new(input);
- 0
+pub fn solve(input: &str, wire_id: &str) -> u16 {
+ let mut circuit = Circuit::new(input);
+ circuit.signal_on(wire_id)
}
#[test]
@@ -52,13 +58,17 @@ impl<'a> Circuit<'a> {
Circuit { connections: connections }
}
+ // TODO This should probably return a Result...
+ // TODO Is there a way for this to not be mutable?
fn signal_on(&mut self, wire_id: &'a str) -> u16 {
- let value = match self.connections.get(&Wire(wire_id)) {
- Some(&Signal::Value(value)) => value,
- Some(&Signal::Wire(Wire(input_wire_id))) => self.signal_on(input_wire_id),
- Some(&Signal::Gate(Gate::And(Wire(a), Wire(b)))) => self.signal_on(a) & self.signal_on(b),
- Some(&Signal::Gate(Gate::Not(Wire(a)))) => !self.signal_on(a),
- _ => 0,
+ let value = match self.connections.get(&Wire(wire_id)).unwrap() {
+ &Signal::Value(value) => value,
+ &Signal::Wire(Wire(input_wire_id)) => self.signal_on(input_wire_id),
+ &Signal::Gate(Gate::And(Wire(a), Wire(b))) => self.signal_on(a) & self.signal_on(b),
+ &Signal::Gate(Gate::LeftShift(Wire(a), i)) => self.signal_on(a) << i,
+ &Signal::Gate(Gate::Not(Wire(a))) => !self.signal_on(a),
+ &Signal::Gate(Gate::Or(Wire(a), Wire(b))) => self.signal_on(a) | self.signal_on(b),
+ &Signal::Gate(Gate::RightShift(Wire(a), i)) => self.signal_on(a) >> i,
};
value
}