def test_example(self):
self.assertEqual(day3.count_trees(EXAMPLE, (1, 1)), 2)
self.assertEqual(day3.count_trees(EXAMPLE, (3, 1)), 7)
self.assertEqual(day3.count_trees(EXAMPLE, (5, 1)), 3)
self.assertEqual(day3.count_trees(EXAMPLE, (7, 1)), 4)
self.assertEqual(day3.count_trees(EXAMPLE, (1, 2)), 2)
self.assertEqual(day3.part_2(EXAMPLE), 336)
def test_part2_2():
assert (part_2(
"R98,U47,R26,D63,R33,U87,L62,D20,R33,U53,R51",
"U98,R91,D20,R16,D67,R40,U7,R15,U6,R7",
) == 410)
def test_part2_2():
assert (part_2("R75,D30,R83,U83,L12,D49,R71,U7,L72",
"U62,R66,U55,R34,D71,R55,D58,R83") == 610)
def test_part2_1():
assert part_2("R8,U5,L5,D3", "U7,R6,D4,L4") == 30
def test_with_input(self):
tree_map = utils.read_input(3)
self.assertEqual(day3.part_2(tree_map), 7540141059)
def test_part_2_real_input(self):
result = part_2(input)
self.assertEqual(result, 1115775000)
def test_part_2_example_input(self):
result = part_2(example_input)
self.assertEqual(result, 336)