def test_add_two_children(self): node = Attribute(5, 'foo') 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_type(self): node = Attribute(5, 'bar') node.type = 'str' rst = node.render_rst() self.assertEqual(rst, ['.. py:attribute:: bar', '', ' (str)', ''])
def test_render_rst_with_child(self): node = Attribute(5, 'bar') child = Node(indent=10, lines=["Description"], parent=node) node.add_child(child) rst = node.render_rst() self.assertEqual(rst, ['.. py:attribute:: bar', '', ' Description', ''])
def test_render_rst_with_type_and_multi_line_description(self): node = Attribute(5, 'bar') node.type = 'str' child = Node(indent=10, lines=["Description1", "Description2"], parent=node) node.add_child(child) rst = node.render_rst() self.assertEqual(rst, ['.. py:attribute:: bar', '', ' (str) Description1', ' Description2', ''])
def test_render_rst_with_children(self): node = Attribute(5, 'bar') child_a = Node(indent=10, lines=["ChildA"], parent=node) node.add_child(child_a) child_b = Node(indent=10, lines=["ChildB"], parent=node) node.add_child(child_b) rst = node.render_rst() self.assertEqual(rst, ['.. py:attribute:: bar', '', ' ChildA', ' ChildB', ''])
def test_render_rst_empty(self): node = Attribute(5, 'bar') rst = node.render_rst() self.assertEqual(rst, ['.. py:attribute:: bar', ''])
def test_add_one_child(self): node = Attribute(5, 'foo') child = Node(parent=node) node.add_child(child) self.assertIs(node.children[0], child)
def test_set_type(self): node = Attribute(5, 'foo') node.type = 'str' self.assertEqual(node.type, 'str')