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

Source Code for Module lxml.tests.test_elementpath

  1  # -*- coding: utf-8 -*- 
  2   
  3  """ 
  4  Tests for the ElementPath implementation. 
  5  """ 
  6   
  7  from __future__ import absolute_import 
  8   
  9  import unittest 
 10  from copy import deepcopy 
 11  from .common_imports import etree, HelperTestCase 
 12   
 13   

 14 -def summarize(elem): 

 15      return elem.tag 

 16   

 17 -def summarize_list(seq): 

 18      return list(map(summarize, seq)) 

 19   

 20 -def normalize_crlf(tree): 

 21      for elem in tree.getiterator(): 
 22          if elem.text: elem.text = elem.text.replace("\r\n", "\n") 
 23          if elem.tail: elem.tail = elem.tail.replace("\r\n", "\n") 

 24   
 25   

 26 -class EtreeElementPathTestCase(HelperTestCase): 

 27      etree = etree 
 28      from lxml import _elementpath 
 29   

 30 -    def test_cache(self): 

 31          self._elementpath._cache.clear() 
 32          el = self.etree.XML(b'<a><b><c/><c/></b></a>') 
 33          self.assertFalse(self._elementpath._cache) 
 34   
 35          self.assertTrue(el.findall('b/c')) 
 36          self.assertEqual(1, len(self._elementpath._cache)) 
 37          self.assertTrue(el.findall('b/c')) 
 38          self.assertEqual(1, len(self._elementpath._cache)) 
 39          self.assertFalse(el.findall('xxx')) 
 40          self.assertEqual(2, len(self._elementpath._cache)) 
 41          self.assertFalse(el.findall('xxx')) 
 42          self.assertEqual(2, len(self._elementpath._cache)) 
 43          self.assertTrue(el.findall('b/c')) 
 44          self.assertEqual(2, len(self._elementpath._cache)) 

 45   

 46 -    def _assert_tokens(self, tokens, path, namespaces=None): 

 47          self.assertEqual(tokens, list(self._elementpath.xpath_tokenizer(path, namespaces))) 

 48   

 49 -    def test_tokenizer(self): 

 50          assert_tokens = self._assert_tokens 
 51          assert_tokens( 
 52              [('/', '')], 
 53              '/', 
 54          ) 
 55          assert_tokens( 
 56              [('.', ''), ('/', ''), ('', 'a'), ('/', ''), ('', 'b'), ('/', ''), ('', 'c')], 
 57              './a/b/c', 
 58          ) 
 59          assert_tokens( 
 60              [('/', ''), ('', 'a'), ('/', ''), ('', 'b'), ('/', ''), ('', 'c')], 
 61              '/a/b/c', 
 62          ) 
 63          assert_tokens( 
 64              [('/', ''), ('', '{nsx}a'), ('/', ''), ('', '{nsy}b'), ('/', ''), ('', 'c')], 
 65              '/x:a/y:b/c', 
 66              {'x': 'nsx', 'y': 'nsy'}, 
 67          ) 
 68          assert_tokens( 
 69              [('/', ''), ('', '{nsx}a'), ('/', ''), ('', '{nsy}b'), ('/', ''), ('', '{nsnone}c')], 
 70              '/x:a/y:b/c', 
 71              {'x': 'nsx', 'y': 'nsy', None: 'nsnone'}, 
 72          ) 

 73   

 75          assert_tokens = self._assert_tokens 
 76          assert_tokens( 
 77              [('', 'a'), ('[', ''), ('', 'b'), (']', '')], 
 78              'a[b]', 
 79          ) 
 80          assert_tokens( 
 81              [('', 'a'), ('[', ''), ('', 'b'), ('=', ''), ('"abc"', ''), (']', '')], 
 82              'a[b="abc"]', 
 83          ) 
 84          assert_tokens( 
 85              [('', 'a'), ('[', ''), ('.', ''), ('', ''), ('=', ''), ('', ''), ('"abc"', ''), (']', '')], 
 86              'a[. = "abc"]', 
 87          ) 

 88   

 89 -    def test_find(self): 

 90          """ 
 91          Test find methods (including xpath syntax). 
 92          Originally copied from 'selftest.py'. 
 93          """ 
 94          elem = etree.XML(""" 
 95          <body> 
 96            <tag class='a'>text</tag> 
 97            <tag class='b' /> 
 98            <section> 
 99              <tag class='b' id='inner'>subtext</tag> 
100            </section> 
101          </body> 
102          """) 
103   
104          self.assertEqual(elem.find("tag").tag, 
105                           'tag') 
106          self.assertEqual(etree.ElementTree(elem).find("tag").tag, 
107                           'tag') 
108          self.assertEqual(elem.find("section/tag").tag, 
109                           'tag') 
110          self.assertEqual(etree.ElementTree(elem).find("section/tag").tag, 
111                           'tag') 
112   
113          self.assertEqual(elem.findtext("tag"), 
114                           'text') 
115          self.assertEqual(elem.findtext("tog"), 
116                           None) 
117          self.assertEqual(elem.findtext("tog", "default"), 
118                           'default') 
119          self.assertEqual(etree.ElementTree(elem).findtext("tag"), 
120                           'text') 
121          self.assertEqual(elem.findtext("section/tag"), 
122                           'subtext') 
123          self.assertEqual(etree.ElementTree(elem).findtext("section/tag"), 
124                           'subtext') 
125   
126          self.assertEqual(summarize_list(elem.findall("tag")), 
127                           ['tag', 'tag']) 
128          self.assertEqual(summarize_list(elem.findall("*")), 
129                           ['tag', 'tag', 'section']) 
130          self.assertEqual(summarize_list(elem.findall(".//tag")), 
131                           ['tag', 'tag', 'tag']) 
132          self.assertEqual(summarize_list(elem.findall("section/tag")), 
133                           ['tag']) 
134          self.assertEqual(summarize_list(elem.findall("section//tag")), 
135                           ['tag']) 
136   
137          self.assertEqual(summarize_list(elem.findall("section/*")), 
138                           ['tag']) 
139          self.assertEqual(summarize_list(elem.findall("section//*")), 
140                           ['tag']) 
141          self.assertEqual(summarize_list(elem.findall("section/.//*")), 
142                           ['tag']) 
143          self.assertEqual(summarize_list(elem.findall("*/*")), 
144                           ['tag']) 
145          self.assertEqual(summarize_list(elem.findall("*//*")), 
146                           ['tag']) 
147          self.assertEqual(summarize_list(elem.findall("*/tag")), 
148                           ['tag']) 
149          self.assertEqual(summarize_list(elem.findall("*/./tag")), 
150                           ['tag']) 
151          self.assertEqual(summarize_list(elem.findall("./tag")), 
152                           ['tag', 'tag']) 
153          self.assertEqual(summarize_list(elem.findall(".//tag")), 
154                           ['tag', 'tag', 'tag']) 
155          self.assertEqual(summarize_list(elem.findall("././tag")), 
156                           ['tag', 'tag']) 
157   
158          self.assertEqual(summarize_list(elem.findall(".//tag[@class]")), 
159                           ['tag', 'tag', 'tag']) 
160          self.assertEqual(summarize_list(elem.findall(".//tag[ @class]")), 
161                           ['tag', 'tag', 'tag']) 
162          self.assertEqual(summarize_list(elem.findall(".//tag[@class ]")), 
163                           ['tag', 'tag', 'tag']) 
164          self.assertEqual(summarize_list(elem.findall(".//tag[  @class  ]")), 
165                           ['tag', 'tag', 'tag']) 
166          self.assertEqual(summarize_list(elem.findall(".//tag[@class='a']")), 
167                           ['tag']) 
168          self.assertEqual(summarize_list(elem.findall('.//tag[@class="a"]')), 
169                           ['tag']) 
170          self.assertEqual(summarize_list(elem.findall(".//tag[@class='b']")), 
171                           ['tag', 'tag']) 
172          self.assertEqual(summarize_list(elem.findall('.//tag[@class="b"]')), 
173                           ['tag', 'tag']) 
174          self.assertEqual(summarize_list(elem.findall('.//tag[@class = "b"]')), 
175                           ['tag', 'tag']) 
176          self.assertEqual(summarize_list(elem.findall(".//tag[@id]")), 
177                           ['tag']) 
178          self.assertEqual(summarize_list(elem.findall(".//tag[@class][@id]")), 
179                           ['tag']) 
180          self.assertEqual(summarize_list(elem.findall(".//section[tag]")), 
181                           ['section']) 
182          self.assertEqual(summarize_list(elem.findall(".//section[element]")), 
183                           []) 
184   
185          self.assertEqual(summarize_list(elem.findall(".//section[tag='subtext']")), 
186                           ['section']) 
187          self.assertEqual(summarize_list(elem.findall(".//section[tag ='subtext']")), 
188                           ['section']) 
189          self.assertEqual(summarize_list(elem.findall(".//section[tag= 'subtext']")), 
190                           ['section']) 
191          self.assertEqual(summarize_list(elem.findall(".//section[tag = 'subtext']")), 
192                           ['section']) 
193          self.assertEqual(summarize_list(elem.findall(".//section[  tag   =   'subtext'  ]")), 
194                           ['section']) 
195          self.assertEqual(summarize_list(elem.findall(".//tag[.='subtext']")), 
196                           ['tag']) 
197          self.assertEqual(summarize_list(elem.findall(".//tag[. ='subtext']")), 
198                           ['tag']) 
199          self.assertEqual(summarize_list(elem.findall('.//tag[.= "subtext"]')), 
200                           ['tag']) 
201          self.assertEqual(summarize_list(elem.findall(".//tag[. = 'subtext']")), 
202                           ['tag']) 
203          self.assertEqual(summarize_list(elem.findall(".//tag[. = 'subtext ']")), 
204                           []) 
205          self.assertEqual(summarize_list(elem.findall(".//tag[.= ' subtext']")), 
206                           []) 
207   
208          self.assertEqual(summarize_list(elem.findall("../tag")), 
209                           []) 
210          self.assertEqual(summarize_list(elem.findall("section/../tag")), 
211                           ['tag', 'tag']) 
212          self.assertEqual(summarize_list(etree.ElementTree(elem).findall("./tag")), 
213                           ['tag', 'tag']) 
214   
215          # FIXME: ET's Path module handles this case incorrectly; this gives 
216          # a warning in 1.3, and the behaviour will be modified in 1.4. 
217          self.assertEqual(summarize_list(etree.ElementTree(elem).findall("/tag")), 
218                           ['tag', 'tag']) 
219   
220          # duplicate section => 2x tag matches 
221          elem[1] = deepcopy(elem[2]) 
222          self.assertEqual(summarize_list(elem.findall(".//section[tag = 'subtext']")), 
223                           ['section', 'section']) 
224          self.assertEqual(summarize_list(elem.findall(".//tag[. = 'subtext']")), 
225                           ['tag', 'tag']) 
226          self.assertEqual(summarize_list(elem.findall(".//tag[@class][@id]")), 
227                           ['tag', 'tag']) 

228   
229   
230  #class ElementTreeElementPathTestCase(EtreeElementPathTestCase): 
231  #    import xml.etree.ElementTree as etree 
232  #    import xml.etree.ElementPath as _elementpath 
233   
234   

235 -def test_suite(): 

236      suite = unittest.TestSuite() 
237      suite.addTests([unittest.makeSuite(EtreeElementPathTestCase)]) 
238      #suite.addTests([unittest.makeSuite(ElementTreeElementPathTestCase)]) 
239      return suite 

240   
241   
242  if __name__ == '__main__': 
243      print('to test use test.py %s' % __file__) 
244

   Home       Trees       Indices       Help   
lxml API
Generated by Epydoc 3.0.1 on Wed Jan 2 20:55:26 2019 http://epydoc.sourceforge.net