def test_part_one(self):
"Test part one example of Nanobots object"
# 1. Create Nanobots object from text
myobj = nanobots.Nanobots(text=aoc_23.from_text(PART_ONE_TEXT))
# 2. Check the part one result
self.assertEqual(myobj.part_one(verbose=False), PART_ONE_RESULT)
def test_part_two(self):
"Test part two example of Nanobots object"
# 1. Create Nanobots object from text
myobj = nanobots.Nanobots(part2=True,
text=aoc_23.from_text(PART_TWO_TEXT))
# 2. Check the part two result
self.assertEqual(myobj.part_two(verbose=False), PART_TWO_RESULT)
def test_empty_init(self):
"Test the default Nanobots creation"
# 1. Create default Nanobots object
myobj = nanobots.Nanobots()
# 2. Make sure it has the default values
self.assertEqual(myobj.part2, False)
self.assertEqual(myobj.text, None)
self.assertEqual(len(myobj.bots), 0)
def part_two(args, input_lines):
"Process part two of the puzzle"
# 1. Create the puzzle solver
solver = nanobots.Nanobots(part2=True, text=input_lines)
# 2. Determine the solution for part two
solution = solver.part_two(verbose=args.verbose, limit=args.limit)
if solution is None:
print("There is no solution")
else:
print("The solution for part two is %s" % (solution))
# 3. Return result
return solution is not None
def test_text_init(self):
"Test the Nanobots object creation from text"
# 1. Create Nanobots object from text
myobj = nanobots.Nanobots(text=aoc_23.from_text(EXAMPLE_TEXT))
# 2. Make sure it has the expected values
self.assertEqual(myobj.part2, False)
self.assertEqual(len(myobj.text), 9)
self.assertEqual(len(myobj.bots), 9)
# 3. Check distances
self.assertEqual(myobj.bots[0].manhattan_distance(myobj.bots[0]), 0)
self.assertEqual(myobj.bots[0].manhattan_distance(myobj.bots[1]), 1)
self.assertEqual(myobj.bots[0].manhattan_distance(myobj.bots[2]), 4)
self.assertEqual(myobj.bots[0].manhattan_distance(myobj.bots[3]), 2)
self.assertEqual(myobj.bots[0].manhattan_distance(myobj.bots[4]), 5)
self.assertEqual(myobj.bots[0].manhattan_distance(myobj.bots[5]), 3)
self.assertEqual(myobj.bots[0].manhattan_distance(myobj.bots[6]), 3)
self.assertEqual(myobj.bots[0].manhattan_distance(myobj.bots[7]), 4)
self.assertEqual(myobj.bots[0].manhattan_distance(myobj.bots[8]), 5)
# 4. Check range
self.assertEqual(myobj.bots[0].in_range(myobj.bots[0]), True)
self.assertEqual(myobj.bots[0].in_range(myobj.bots[1]), True)
self.assertEqual(myobj.bots[0].in_range(myobj.bots[2]), True)
self.assertEqual(myobj.bots[0].in_range(myobj.bots[3]), True)
self.assertEqual(myobj.bots[0].in_range(myobj.bots[4]), False)
self.assertEqual(myobj.bots[0].in_range(myobj.bots[5]), True)
self.assertEqual(myobj.bots[0].in_range(myobj.bots[6]), True)
self.assertEqual(myobj.bots[0].in_range(myobj.bots[7]), True)
self.assertEqual(myobj.bots[0].in_range(myobj.bots[8]), False)
# 4. Check strongest signal
self.assertEqual(myobj.largest_signal_radius(), myobj.bots[0])
# 5. Check number in range
self.assertEqual(myobj.bots_in_range(myobj.bots[0]), 7)