from itertools import *
import benchbase
from benchbase import onlylib, children, nochange
############################################################
# Benchmarks
############################################################
class XPathBenchMark(benchbase.TreeBenchMark):
@nochange
@onlylib('lxe')
@children
def bench_xpath_class(self, children):
xpath = self.etree.XPath("./*[1]")
for child in children:
xpath(child)
@nochange
@onlylib('lxe')
@children
def bench_xpath_class_repeat(self, children):
for child in children:
xpath = self.etree.XPath("./*[1]")
xpath(child)
@nochange
@onlylib('lxe')
def bench_xpath_element(self, root):
xpath = self.etree.XPathElementEvaluator(root)
for child in root:
xpath.evaluate("./*[1]")
@nochange
@onlylib('lxe')
@children
def bench_xpath_method(self, children):
for child in children:
child.xpath("./*[1]")
@nochange
@onlylib('lxe')
@children
def bench_multiple_xpath_or(self, children):
xpath = self.etree.XPath(".//p:a00001|.//p:b00001|.//p:c00001",
namespaces={'p':'cdefg'})
for child in children:
xpath(child)
@nochange
@onlylib('lxe')
@children
def bench_multiple_iter_tag(self, children):
for child in children:
list(child.iter("{cdefg}a00001"))
list(child.iter("{cdefg}b00001"))
list(child.iter("{cdefg}c00001"))
@nochange
@onlylib('lxe')
@children
def bench_xpath_old_extensions(self, children):
def return_child(_, elements):
if elements:
return elements[0][0]
else:
return ()
extensions = {("test", "child") : return_child}
xpath = self.etree.XPath("t:child(.)", namespaces={"t":"test"},
extensions=extensions)
for child in children:
xpath(child)
@nochange
@onlylib('lxe')
@children
def bench_xpath_extensions(self, children):
def return_child(_, elements):
if elements:
return elements[0][0]
else:
return ()
self.etree.FunctionNamespace("testns")["t"] = return_child
try:
xpath = self.etree.XPath("test:t(.)", namespaces={"test":"testns"})
for child in children:
xpath(child)
finally:
del self.etree.FunctionNamespace("testns")["t"]
if __name__ == '__main__':
benchbase.main(XPathBenchMark)