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
     74 -    def test_tokenizer_predicates(self):
     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