def testAdd1(self): tree = trees.Trees() tree.add(None, 'R') tree.add('R', 'A') self.__ae('R', tree, 1, 4) self.__ae('A', tree, 2, 3)
def testRemove0(self): tree = trees.Trees() tree.add(None, 'R') tree.add('R', 'A') tree.remove('A') self.__ae('R', tree, 1, 2)
def test_text_init(self): "Test the Trees object creation from text" # 1. Create Map object from text myobj = trees.Trees(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), 11) self.assertEqual(len(myobj.trees), 11) self.assertEqual(myobj.rows, 11) self.assertEqual(myobj.cols, 11) # 3. Test methods self.assertEqual(myobj.is_tree((0, 0)), False) self.assertEqual(myobj.is_tree((1, 0)), False) self.assertEqual(myobj.is_tree((2, 0)), True) self.assertEqual(myobj.is_tree((3, 1)), False) self.assertEqual(myobj.is_tree((6, 2)), True) self.assertEqual(myobj.is_tree((9, 3)), False) self.assertEqual(myobj.is_tree((12, 4)), True) self.assertEqual(myobj.is_tree((15, 5)), True) self.assertEqual(myobj.is_tree((18, 6)), False) self.assertEqual(myobj.is_tree((21, 7)), True) self.assertEqual(myobj.is_tree((24, 8)), True) self.assertEqual(myobj.is_tree((27, 9)), True) self.assertEqual(myobj.is_tree((30, 10)), True) self.assertEqual(myobj.is_tree((33, 11)), False) self.assertEqual(myobj.is_tree((36, 12)), False) self.assertEqual(myobj.is_last((0, 0)), False) self.assertEqual(myobj.is_last((9, 3)), False) self.assertEqual(myobj.is_last((27, 9)), False) self.assertEqual(myobj.is_last((30, 10)), True) self.assertEqual(myobj.is_last((33, 11)), True)
def testAdd3(self): tree = trees.Trees() tree.add(None, 'R') tree.add('R', 'A') tree.add('A', 'A1') self.__ae('R', tree, 1, 6) self.__ae('A', tree, 2, 5) self.__ae('A1', tree, 3, 4)
def testAdd4(self): tree = trees.Trees() tree.add(None, 'R') tree.add('R', 'A') tree.add('A', 'A1') tree.add('A', 'A2') self.__ae('R', tree, 1, 8) self.__ae('A', tree, 2, 7) self.__ae('A1', tree, 3, 4) self.__ae('A2', tree, 5, 6)
def testRemove1(self): tree = trees.Trees() tree.add(None, 'R') tree.add('R', 'A') tree.add('A', 'A1') tree.add('A', 'A2') tree.remove('A1') self.__ae('R', tree, 1, 6) self.__ae('A', tree, 2, 5) self.__ae('A2', tree, 3, 4)
def test_empty_init(self): "Test the default Trees creation" # 1. Create default Map object myobj = trees.Trees() # 2. Make sure it has the default values self.assertEqual(myobj.part2, False) self.assertEqual(myobj.text, None) self.assertEqual(myobj.trees, None) self.assertEqual(myobj.rows, 0) self.assertEqual(myobj.cols, 0)
def __init__(self, text=None, part2=False): # 1. Set the initial values self.part2 = part2 self.text = text self.loc = (0, 0) self.delta = (3, 1) self.trees = None # 2. Process text (if any) if text is not None and len(text) > 0: self.trees = trees.Trees(text=text, part2=part2)
def testAdd5(self): tree = trees.Trees() tree.add(None, 'R') tree.add('R', 'A') tree.add('A', 'A1') tree.add('A', 'A2') tree.add('R', 'B') self.__ae('R', tree, 1, 10) self.__ae('A', tree, 2, 7) self.__ae('A1', tree, 3, 4) self.__ae('A2', tree, 5, 6) self.__ae('B', tree, 8, 9)
def testRemove3(self): tree = trees.Trees() tree.add(None, 'R') tree.add('R', 'A') tree.add('A', 'A1') tree.add('A', 'A2') tree.add('R', 'B') tree.add('B', 'B1') tree.add('B', 'B2') tree.remove('A') self.__ae('R', tree, 1, 12) self.__ae('A1', tree, 2, 3) self.__ae('A2', tree, 4, 5) self.__ae('B', tree, 6, 11) self.__ae('B1', tree, 7, 8) self.__ae('B2', tree, 9, 10)
def testAdd9(self): tree = trees.Trees() tree.add(None, 'R') tree.add('R', 'A') tree.add('A', 'A1') tree.add('A', 'A2') tree.add('R', 'B') tree.add('B', 'B1') tree.add('B', 'B2') tree.add('A', 'A3') self.__ae('R', tree, 1, 16) self.__ae('A', tree, 2, 9) self.__ae('A1', tree, 3, 4) self.__ae('A2', tree, 5, 6) self.__ae('B', tree, 10, 15) self.__ae('B1', tree, 11, 12) self.__ae('B2', tree, 13, 14) self.__ae('A3', tree, 7, 8)
def testAdd10(self): tree = trees.Trees() tree.add(None, 'R') tree.add('R', 'A') tree.add('A', 'A1') tree.add('A', 'A2') tree.add('R', 'B') tree.add('B', 'B1') tree.add('B', 'B2') tree.add('A', 'A3') tree.add('A2', 'A21') self.__ae('R', tree, 1, 18) self.__ae('A', tree, 2, 11) self.__ae('A1', tree, 3, 4) self.__ae('A2', tree, 5, 8) self.__ae('B', tree, 12, 17) self.__ae('B1', tree, 13, 14) self.__ae('B2', tree, 15, 16) self.__ae('A3', tree, 9, 10) self.__ae('A21', tree, 6, 7)
def testAdd0(self): tree = trees.Trees() tree.add(None, 'R') self.__ae('R', tree, 1, 2)