def test_empty_init(self): "Test the default Triangles creation" # 1. Create default Triangles object myobj = triangles.Triangles() # 2. Make sure it has the default values self.assertEqual(myobj.part2, False) self.assertEqual(myobj.text, []) # 3. Check methods self.assertEqual(myobj.check_sides([5, 10, 25]), False) self.assertEqual(myobj.check_sides([10, 5, 25]), False) self.assertEqual(myobj.check_sides([7, 9, 13]), True) self.assertEqual(myobj.check_sides([9, 7, 13]), True) self.assertEqual(myobj.check_sides([9, 7, 13, 1]), False) self.assertEqual(myobj.check_sides([]), False) self.assertEqual(myobj.check_sides([15, 4, 8]), False) self.assertEqual(myobj.check_row(" 5 10 25"), False) self.assertEqual(myobj.check_row(" 7 9 13"), True) self.assertEqual(myobj.check_row(" 15 4 8"), False) self.assertEqual(myobj.check_row("810 783 545"), True) self.assertEqual(myobj.check_row(" 84 607 556"), True) self.assertEqual(myobj.check_row("679 225 626"), True) self.assertEqual(myobj.check_row("910 425 616"), True) self.assertEqual(myobj.check_row(" 10 616 626"), False) self.assertEqual(myobj.check_row("149 901 883"), True)
def test_part_one(self): "Test part one example of Triangles object" # 1. Create Triangles object from text myobj = triangles.Triangles(text=aoc_03.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 Triangles object" # 1. Create Triangles object from text myobj = triangles.Triangles(part2=True, text=aoc_03.from_text(PART_TWO_TEXT)) # 2. Check the part two result self.assertEqual(myobj.part_two(verbose=False), PART_TWO_RESULT)
def test_text_init(self): "Test the Triangles object creation from text" # 1. Create Triangles object from text myobj = triangles.Triangles(text=aoc_03.from_text(EXAMPLE_TEXT)) # 2. Make sure it has the expected values self.assertEqual(myobj.part2, False) self.assertEqual(len(myobj.text), 3) # 3. Check methods self.assertEqual(myobj.count_triangles(), 1)
def part_two(args, input_lines): "Process part two of the puzzle" # 1. Create the puzzle solver solver = triangles.Triangles(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_pivot(self): ts = triangles.Triangles() ts.readFile("input2016_03a.txt") ts.pivot() self.assertEqual(None, ts.count())
def test_readTriangles(self): ts = triangles.Triangles() ts.readFile("input2016_03a.txt") self.assertEqual(917, ts.count())