old_contents: #!/usr/bin/env python
  
  import os
  import re
  import unittest
  from xml_driver import XMLElement, XMLHandler
  from xml.sax import make_parser, handler
  
  # Directory of test files
  xml_files = [x for x in os.listdir('test_xml_files')
               if re.match(r"2012_\d.xml", x) != None] # Match fixtures
  
  parsed_xml = []
  for xf in xml_files:
      parser = make_parser()
      xmlhandler = XMLHandler()
      parser.setContentHandler(xmlhandler)
      parser.setFeature(handler.feature_external_ges, False)
      parser.parse('test_xml_files/'+xf)
      parsed_xml.append(xmlhandler.root)
  
  class Test_XMLElement(unittest.TestCase):
      
      def setUp(self):
          # sanity check
          self.assertTrue(xml_files)
  
      def test_basic_xml_tag_counts(self):
          xmlhandler = XMLHandler()
          parser.setContentHandler(xmlhandler)
          parser.setFeature(handler.feature_external_ges, False)
          parser.parse('test_xml_files/basic.xml')
          self.assertTrue(xmlhandler.root)
          root = xmlhandler.root
          self.assertTrue(len(root.a) == 1)
          self.assertTrue(len(root.a.b) == 2)
          self.assertTrue(len(root.a.b.c) == 3)
          self.assertTrue(len(root.a.b.d) == 2)
  
  
  
  unittest.main()
  
new_contents: #!/usr/bin/env python
  
  import os
  import re
  import unittest
  from xml_driver import XMLElement, XMLHandler
  from xml.sax import make_parser, handler
  
  # Directory of test files
  xml_files = [x for x in os.listdir('test_xml_files')
               if re.match(r"2012_\d.xml", x) != None] # Match fixtures
  
  parsed_xml = []
  for xf in xml_files:
      parser = make_parser()
      xmlhandler = XMLHandler()
      parser.setContentHandler(xmlhandler)
      parser.setFeature(handler.feature_external_ges, False)
      parser.parse('test_xml_files/'+xf)
      parsed_xml.append(xmlhandler.root)
  
  class Test_XMLElement(unittest.TestCase):
      
      def setUp(self):
          # sanity check
          self.assertTrue(xml_files)
  
      def test_basic_xml_tag_counts(self):
          xmlhandler = XMLHandler()
          parser.setContentHandler(xmlhandler)
          parser.setFeature(handler.feature_external_ges, False)
          parser.parse('test_xml_files/basic.xml')
          self.assertTrue(xmlhandler.root)
          root = xmlhandler.root
          self.assertTrue(len(root.a) == 1)
          self.assertTrue(len(root.a.b) == 2)
          self.assertTrue(len(root.a.b.c) == 3)
          self.assertTrue(len(root.a.b.d) == 2)
  
      def test_basic_xml_tag_contents(self):
          xmlhandler = XMLHandler()
          parser.setContentHandler(xmlhandler)
          parser.setFeature(handler.feature_external_ges, False)
          parser.parse('test_xml_files/basic.xml')
          root = xmlhandler.root
          self.assertTrue(root.a.b.c[0].content  == 'hello', \
              "{0} should be {1}".format(root.a.b.c[0].content, 'hello'))
          self.assertTrue(root.a.b.c[1].content  == 'world', \
              "{0} should be {1}".format(root.a.b.c[1].content, 'world'))
          self.assertTrue(root.a.b.c[2].content  == '3', \
              "{0} should be {1}".format(root.a.b.c[2].content, '3'))
          self.assertTrue(root.a.b.d[0].content  == '1', \
              "{0} should be {1}".format(root.a.b.c[0].content, '1'))
          self.assertTrue(root.a.b.d[1].content  == '2', \
              "{0} should be {1}".format(root.a.b.c[1].content, '2'))
          
  
  
  unittest.main()
  
subject: add xml content test