def test_add_two_children(self): node = Usage(5) child0 = Node(parent=node) child1 = Node(parent=node) node.add_child(child0) node.add_child(child1) self.assertIs(node.children[0], child0) self.assertIs(node.children[1], child1)
def test_render_rst_with_indented_code(self): node = Usage(5) child_a = Node(indent=10, lines=['for i in range(100):'], parent=node) node.add_child(child_a) child_b = Node(indent=14, lines=['print(i)'], parent=node) node.add_child(child_b) rst = node.render_rst() self.assertEqual(rst, [' .. rubric:: Usage:', '', ' .. code-block:: python', '', ' for i in range(100):', ' print(i)', ''])
def test_render_rst_with_code(self): node = Usage(7) child_a = Node(indent=10, lines=['print("Hello, World!")'], parent=node) node.add_child(child_a) child_b = Node(indent=10, lines=['print("Reticulating splines!")'], parent=node) node.add_child(child_b) rst = node.render_rst() self.assertEqual(rst, [' .. rubric:: Usage:', '', ' .. code-block:: python', '', ' print("Hello, World!")', ' print("Reticulating splines!")', ''])
def test_render_rst_empty(self): node = Usage(5) rst = node.render_rst() self.assertEqual(rst, [])
def test_add_one_child(self): node = Usage(5) child = Node(parent=node) node.add_child(child) self.assertIs(node.children[0], child)