Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add function to measure performance

Browse Source
This commit is contained in:
Kovid Goyal 2013-10-28 13:22:54 +05:30
parent 45c2211c4b
commit ddec629bd7
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/calibre/ebooks/oeb/polish/tests/parsing.py
View File

@ -6,6 +6,8 @@ from __future__ import (unicode_literals, division, absolute_import,
__license__ = 'GPL v3' __license__ = 'GPL v3'
__copyright__ = '2013, Kovid Goyal <kovid at kovidgoyal.net>' __copyright__ = '2013, Kovid Goyal <kovid at kovidgoyal.net>'
from functools import partial
from lxml import etree from lxml import etree
from html5lib.constants import cdataElements, rcdataElements from html5lib.constants import cdataElements, rcdataElements
@ -32,7 +34,7 @@ def namespaces(test, parse_function):
def match_and_prefix(root, xpath, prefix, err=''): def match_and_prefix(root, xpath, prefix, err=''):
matches = XPath(xpath)(root) matches = XPath(xpath)(root)
ae(len(matches), 1, err) ae(len(matches), 1, err)
ae(matches[0].prefix, prefix) ae(matches[0].prefix, prefix, err)
markup = ''' <html xmlns="{xhtml}"><head><body id="test"></html> '''.format(xhtml=XHTML_NS) markup = ''' <html xmlns="{xhtml}"><head><body id="test"></html> '''.format(xhtml=XHTML_NS)
root = parse_function(markup) root = parse_function(markup)
@ -175,3 +177,20 @@ class ParsingTests(BaseTest):
for i, (k, v) in enumerate(root.xpath('//*[local-name()="%s"]' % tag)[0].items()): for i, (k, v) in enumerate(root.xpath('//*[local-name()="%s"]' % tag)[0].items()):
self.assertEqual(i+1, int(v)) self.assertEqual(i+1, int(v))
def timing():
import time, sys
from calibre.ebooks.chardet import xml_to_unicode
from html5lib import parse as vanilla
filename = sys.argv[-1]
with open(filename, 'rb') as f:
raw = f.read()
raw = xml_to_unicode(raw)[0]
for name, f in (('calibre', partial(parse, line_numbers=False)), ('html5lib', vanilla), ('calibre-old', html5_parse)):
timings = []
for i in xrange(5):
st = time.time()
f(raw)
timings.append(time.time() - st)
avg = sum(timings)/len(timings)
print ('Average time for %s: %.2g' % (name, avg))