Home       Trees       Indices       Help   
lxml API
Package lxml :: Package tests :: Module test_doctestcompare
[hide private]
[frames] | no frames]

Source Code for Module lxml.tests.test_doctestcompare

  1  import unittest 
  2   
  3  from lxml import etree 
  4  from lxml.tests.common_imports import HelperTestCase 
  5  from lxml.doctestcompare import LXMLOutputChecker, PARSE_HTML, PARSE_XML 
  6   
  7   

  8 -class DummyInput: 

  9 -    def __init__(self, **kw): 

 10          for name, value in kw.items(): 
 11              setattr(self, name, value) 

 12   
 13   

 14 -def indent(elem, level=0): 

 15      i = "\n" + level*"  " 
 16      if len(elem): 
 17          if not elem.text or not elem.text.strip(): 
 18              elem.text = i + "  " 
 19          if not elem.tail or not elem.tail.strip(): 
 20              elem.tail = i 
 21          for elem in elem: 
 22              indent(elem, level+1) 
 23          if not elem.tail or not elem.tail.strip(): 
 24              elem.tail = i 
 25      else: 
 26          if level and (not elem.tail or not elem.tail.strip()): 
 27              elem.tail = i 

 28   
 29   

 30 -class DoctestCompareTest(HelperTestCase): 

 31      _checker = LXMLOutputChecker() 
 32   

 33 -    def compare(self, want, got, html=False): 

 34          if html: 
 35              options = PARSE_HTML 
 36          else: 
 37              options = PARSE_XML 
 38   
 39          parse = self._checker.get_parser(want, got, options) 
 40          want_doc = parse(want) 
 41          got_doc = parse(got) 
 42          return self._checker.collect_diff( 
 43              want_doc, got_doc, html, indent=0).lstrip() 

 44   

 45 -    def assert_diff(self, want, got, diff, html=False): 

 46          self.assertEqual(self.compare(want, got, html), diff) 

 47   

 48 -    def assert_nodiff(self, want, got, html=False): 

 49          root = etree.fromstring(want) 
 50          root.tail = '\n' 
 51          indent(root) 
 52          diff = etree.tostring( 
 53              root, encoding='unicode', method=html and 'html' or 'xml') 
 54          self.assert_diff(want, got, diff, html=html) 

 55   

 56 -    def test_equal_input(self): 

 57          self.assert_nodiff( 
 58              '<p title="expected">Expected</p>', 
 59              '<p title="expected">Expected</p>') 

 60   

 61 -    def test_differing_tags(self): 

 62          self.assert_diff( 
 63              '<p title="expected">Expected</p>', 
 64              '<b title="expected">Expected</b>', 
 65              '<p (got: b) title="expected">Expected</p (got: b)>\n') 

 66   

 68          self.assert_diff( 
 69              '<p title="expected">Expected</p>', 
 70              '<P title="expected">Expected</P>', 
 71              '<p (got: P) title="expected">Expected</p (got: P)>\n') 

 72   

 74          self.assert_nodiff( 
 75              '<html><body><p title="expected">Expected</p></body></html>', 
 76              '<HTML><BODY><P title="expected">Expected</P></BODY></HTML>', 
 77              html=True) 

 78   

 80          self.assert_diff( 
 81              '<p title="expected">Expected</p>', 
 82              '<p title="actual">Actual</p>', 
 83              '<p title="expected (got: actual)">Expected (got: Actual)</p>\n') 

 84   

 85 -    def test_extra_children(self): 

 86          # https://bugs.launchpad.net/lxml/+bug/1238503 
 87          self.assert_diff( 
 88              '<p><span>One</span></p>', 
 89              '<p><span>One</span><b>Two</b><em>Three</em></p>', 
 90              '<p>\n' 
 91              '  <span>One</span>\n' 
 92              '  +<b>Two</b>\n' 
 93              '  +<em>Three</em>\n' 
 94              '</p>\n') 

 95   

 96 -    def test_missing_children(self): 

 97          self.assert_diff( 
 98              '<p><span>One</span><b>Two</b><em>Three</em></p>', 
 99              '<p><span>One</span></p>', 
100              '<p>\n' 
101              '  <span>One</span>\n' 
102              '  -<b>Two</b>\n' 
103              '  -<em>Three</em>\n' 
104              '</p>\n') 

105   

106 -    def test_extra_attributes(self): 

107          self.assert_diff( 
108              '<p><span class="foo">Text</span></p>', 
109              '<p><span class="foo" id="bar">Text</span></p>', 
110              '<p>\n' 
111              '  <span class="foo" +id="bar">Text</span>\n' 
112              '</p>\n') 

113   

114 -    def test_missing_attributes(self): 

115          self.assert_diff( 
116              '<p><span class="foo" id="bar">Text</span></p>', 
117              '<p><span class="foo">Text</span></p>', 
118              '<p>\n' 
119              '  <span class="foo" -id="bar">Text</span>\n' 
120              '</p>\n') 

121   
122   

123 -def test_suite(): 

124      suite = unittest.TestSuite() 
125      suite.addTests([unittest.makeSuite(DoctestCompareTest)]) 
126      return suite 

127   
128   
129  if __name__ == '__main__': 
130      unittest.main() 
131

   Home       Trees       Indices       Help   
lxml API
Generated by Epydoc 3.0.1 on Tue Mar 26 13:39:43 2019 http://epydoc.sourceforge.net