| 1 | # -*- coding: utf-8 -*- |
|---|
| 2 | # |
|---|
| 3 | # Copyright (C) 2006-2008 Edgewall Software |
|---|
| 4 | # All rights reserved. |
|---|
| 5 | # |
|---|
| 6 | # This software is licensed as described in the file COPYING, which |
|---|
| 7 | # you should have received as part of this distribution. The terms |
|---|
| 8 | # are also available at http://genshi.edgewall.org/wiki/License. |
|---|
| 9 | # |
|---|
| 10 | # This software consists of voluntary contributions made by many |
|---|
| 11 | # individuals. For the exact contribution history, see the revision |
|---|
| 12 | # history and logs, available at http://genshi.edgewall.org/log/. |
|---|
| 13 | |
|---|
| 14 | |
|---|
| 15 | try: |
|---|
| 16 | from os import times |
|---|
| 17 | def time_func(): |
|---|
| 18 | tup = times() |
|---|
| 19 | #just user time |
|---|
| 20 | return tup[0] # + tup[1] |
|---|
| 21 | except ImportError: |
|---|
| 22 | from time import time as time_func |
|---|
| 23 | |
|---|
| 24 | from genshi.core import START, END |
|---|
| 25 | from genshi.path import Path |
|---|
| 26 | from genshi.input import XML |
|---|
| 27 | |
|---|
| 28 | def benchmark(f, acurate_time=1): |
|---|
| 29 | """Checks how much time does function f work. It runs it as |
|---|
| 30 | many times as needed for avoiding inaccuracy""" |
|---|
| 31 | |
|---|
| 32 | runs = 1 |
|---|
| 33 | while True: |
|---|
| 34 | start_time = time_func() |
|---|
| 35 | for _ in xrange(runs): |
|---|
| 36 | f() |
|---|
| 37 | dt = time_func() - start_time |
|---|
| 38 | if dt >= acurate_time: |
|---|
| 39 | break |
|---|
| 40 | runs *= 2 |
|---|
| 41 | return dt / runs |
|---|
| 42 | |
|---|
| 43 | def spell(t): |
|---|
| 44 | """Returns spelled representation of time""" |
|---|
| 45 | units = [(0.000001, 'microsecond', 'microseconds'), |
|---|
| 46 | (0.001, 'milisecond', 'miliseconds'), |
|---|
| 47 | (1, 'second', 'seconds'), |
|---|
| 48 | (60, 'minute', 'minutes'), |
|---|
| 49 | (60*60, 'hour', 'hours'), |
|---|
| 50 | ] |
|---|
| 51 | i = 0 |
|---|
| 52 | at = abs(t) |
|---|
| 53 | while i + 1 < len(units) and at >= units[i + 1][0]: |
|---|
| 54 | i += 1 |
|---|
| 55 | t /= units[i][0] |
|---|
| 56 | if t >= 2: |
|---|
| 57 | name = units[i][2] |
|---|
| 58 | else: |
|---|
| 59 | name = units[i][1] |
|---|
| 60 | return "%f %s"%(t, name) |
|---|
| 61 | |
|---|
| 62 | def test_paths_in_streams(exprs, streams, test_strategies=False): |
|---|
| 63 | for expr in exprs: |
|---|
| 64 | print "Testing path %r" % expr |
|---|
| 65 | for stream, sname in streams: |
|---|
| 66 | print '\tRunning on "%s" example:' % sname |
|---|
| 67 | |
|---|
| 68 | path = Path(expr) |
|---|
| 69 | def f(): |
|---|
| 70 | for e in path.select(stream): |
|---|
| 71 | pass |
|---|
| 72 | t = spell(benchmark(f)) |
|---|
| 73 | print "\t\tselect:\t\t%s" % t |
|---|
| 74 | |
|---|
| 75 | def f(): |
|---|
| 76 | path = Path(expr) |
|---|
| 77 | for e in path.select(stream): |
|---|
| 78 | pass |
|---|
| 79 | t = spell(benchmark(f)) |
|---|
| 80 | print "\t\tinit + select:\t%s" % t |
|---|
| 81 | |
|---|
| 82 | if test_strategies and len(path.paths) == 1: |
|---|
| 83 | from genshi.path import GenericStrategy, SingleStepStrategy, \ |
|---|
| 84 | SimplePathStrategy |
|---|
| 85 | from genshi.tests.path import FakePath |
|---|
| 86 | strategies = (GenericStrategy, SingleStepStrategy, |
|---|
| 87 | SimplePathStrategy) |
|---|
| 88 | for strategy in strategies: |
|---|
| 89 | if not strategy.supports(path.paths[0]): |
|---|
| 90 | continue |
|---|
| 91 | print "\t\t%s Strategy"%strategy.__name__ |
|---|
| 92 | fp = FakePath(strategy(path.paths[0])) |
|---|
| 93 | def f(): |
|---|
| 94 | for e in fp.select(stream): |
|---|
| 95 | pass |
|---|
| 96 | t = spell(benchmark(f)) |
|---|
| 97 | print "\t\t\tselect:\t\t%s"%t |
|---|
| 98 | |
|---|
| 99 | |
|---|
| 100 | def test_documents(test_strategies=False): |
|---|
| 101 | streams = [] |
|---|
| 102 | |
|---|
| 103 | s = XML("""\ |
|---|
| 104 | <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="pl" xmlns:py="http://genshi.edgewall.org/" py:strip="" lang="en"> |
|---|
| 105 | <head> |
|---|
| 106 | <meta http-equiv="Content-Type" content="text/html; charset=utf-8" /> |
|---|
| 107 | <title>Foo</title> |
|---|
| 108 | </head> |
|---|
| 109 | <body> |
|---|
| 110 | <h1>Hello</h1> |
|---|
| 111 | </body> |
|---|
| 112 | </html> |
|---|
| 113 | """) |
|---|
| 114 | streams.append((s, "small document")) |
|---|
| 115 | |
|---|
| 116 | s = XML("""\ |
|---|
| 117 | <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="pl" xmlns:py="http://genshi.edgewall.org/" py:strip="" lang="en"> |
|---|
| 118 | <head> |
|---|
| 119 | <meta http-equiv="Content-Type" content="text/html; charset=utf-8" /> |
|---|
| 120 | <title>Foo</title> |
|---|
| 121 | </head> |
|---|
| 122 | <body> |
|---|
| 123 | <h1>Hello</h1> |
|---|
| 124 | <div id="splash"> |
|---|
| 125 | <ul> |
|---|
| 126 | <li><a class="b1" href="http://genshi.edgewall.org/"> |
|---|
| 127 | <strong>Genshi</strong> |
|---|
| 128 | Python toolkit for generating output for the web</a></li> |
|---|
| 129 | <li><a class="b2" href="http://babel.edgewall.org/"> |
|---|
| 130 | <strong>Babel</strong> |
|---|
| 131 | Python library for I18n/L10n in web applications</a></li> |
|---|
| 132 | <li><a class="b3" href="http://bitten.edgewall.org/"> |
|---|
| 133 | <strong>Bitten</strong> |
|---|
| 134 | Continuous integration plugin for Trac</a></li> |
|---|
| 135 | <li><a class="b4" href="http://posterity.edgewall.org/"> |
|---|
| 136 | <strong>Posterity</strong> |
|---|
| 137 | Web-based email system</a></li> |
|---|
| 138 | </ul> |
|---|
| 139 | <div id="trac-splash"> |
|---|
| 140 | <a href="http://trac.edgewall.org/"> |
|---|
| 141 | <strong>Trac</strong> Web-based lightweight project management |
|---|
| 142 | system |
|---|
| 143 | </a> |
|---|
| 144 | </div> |
|---|
| 145 | </div> |
|---|
| 146 | </body> |
|---|
| 147 | </html> |
|---|
| 148 | """) |
|---|
| 149 | streams.append((s, "big document")) |
|---|
| 150 | |
|---|
| 151 | paths = [ |
|---|
| 152 | '.', |
|---|
| 153 | '*|text()', |
|---|
| 154 | 'html', |
|---|
| 155 | 'html[@lang="en"]', |
|---|
| 156 | 'html/body/h1/text()', |
|---|
| 157 | 'html/body/div/a/@href', |
|---|
| 158 | 'html/body/div[@id="splash"]/a[@class="b4"]/strong/text()', |
|---|
| 159 | 'descendant-or-self::text()', |
|---|
| 160 | 'descendant-or-self::h1/text()', |
|---|
| 161 | ] |
|---|
| 162 | test_paths_in_streams(paths, streams, test_strategies) |
|---|
| 163 | |
|---|
| 164 | if __name__ == '__main__': |
|---|
| 165 | from sys import argv |
|---|
| 166 | if "--strategies" in argv: |
|---|
| 167 | test_strategies = True |
|---|
| 168 | else: |
|---|
| 169 | test_strategies = False |
|---|
| 170 | test_documents(test_strategies) |
|---|
