def test_part1_example(self):
with open("../inputs/day7_example.txt") as ex_file:
bags = day7.bags_parse(ex_file.read())
print(bags)
self.assertTrue(
day7.bags_contain(bags, "vibrant plum", "faded blue"))
self.assertEqual(4, day7.bags_can_contain(bags, "shiny gold"))
def test_bags_can_contain(self):
bags = day7.bags_parse("blue bags contain 1 red bag.\n"
"red bags contain 1 blue bag.\n"
"green bags contain no other bags.\n")
self.assertEqual(2, day7.bags_can_contain(bags, "red"))
self.assertEqual(2, day7.bags_can_contain(bags, "blue"))
def test_parse_bags(self):
bags = day7.bags_parse("blue bags contain 1 red bag.")
self.assertTrue(day7.bags_contain(bags, "blue", "red"))
def test_part2(self):
with open("../inputs/day7.txt") as day7_file:
bags = day7.bags_parse(day7_file.read())
self.assertEqual(3765,
day7.bags_count_contains(bags, "shiny gold"))
def test_bags_count_contains(self):
with open("../inputs/day7_example.txt") as ex_file:
bags = day7.bags_parse(ex_file.read())
self.assertEqual(32, day7.bags_count_contains(bags, "shiny gold"))
def test_part1(self):
with open("../inputs/day7.txt") as day7_file:
bags = day7.bags_parse(day7_file.read())
self.assertEqual(261, day7.bags_can_contain(bags, "shiny gold"))