[Docutils-checkins] docutils/test test_nodes.py,1.1.1.1,1.2

[<go...@us...>]

Update of /cvsroot/docutils/docutils/test
In directory usw-pr-cvs1:/tmp/cvs-serv907/docutils/test

Modified Files:
	test_nodes.py 
Log Message:
updated

Index: test_nodes.py
===================================================================
RCS file: /cvsroot/docutils/docutils/test/test_nodes.py,v
retrieving revision 1.1.1.1
retrieving revision 1.2
diff -C2 -d -r1.1.1.1 -r1.2
*** test_nodes.py	20 Apr 2002 03:03:04 -0000	1.1.1.1
--- test_nodes.py	5 May 2002 15:48:05 -0000	1.2
***************
*** 12,16 ****
  
  import unittest
! from DocutilsTestSupport import nodes
  
  debug = 0
--- 12,16 ----
  
  import unittest
! from DocutilsTestSupport import nodes, utils
  
  debug = 0
***************
*** 78,81 ****
--- 78,112 ----
      more
  """)
+ 
+ 
+ class TreeCopyVisitorTests(unittest.TestCase):
+ 
+     def setUp(self):
+         document = utils.new_document()
+         document += nodes.paragraph('', 'Paragraph 1.')
+         blist = nodes.bullet_list()
+         for i in range(1, 6):
+             item = nodes.list_item()
+             for j in range(1, 4):
+                 item += nodes.paragraph('', 'Item %s, paragraph %s.' % (i, j))
+             blist += item
+         document += blist
+         self.document = document
+ 
+     def compare_trees(self, one, two):
+         self.assertEquals(one.__class__, two.__class__)
+         self.assertNotEquals(id(one), id(two))
+         children1 = one.getchildren()
+         children2 = two.getchildren()
+         self.assertEquals(len(children1), len(children2))
+         for i in range(len(children1)):
+             self.compare_trees(children1[i], children2[i])
+ 
+     def test_copy_whole(self):
+         visitor = nodes.TreeCopyVisitor(self.document)
+         self.document.walkabout(visitor)
+         newtree = visitor.get_tree_copy()
+         self.assertEquals(self.document.pformat(), newtree.pformat())
+         self.compare_trees(self.document, newtree)