Adapt the speed tests for PyPy.

Author: SimonSapin

tinycss/tests/speed.py

@@ -81,25 +81,45 @@ def parse_cssutils():
 
 
 defcheck_consistency():
+result=parse_python()
 #import pprint
-#pprint.pprint(parse_python())
-result=parse_cython()
+#pprint.pprint(result)
 assertlen(result) >0
-assertparse_python() ==result
+iftokenizer.cython_tokenize_flat:
+assertparse_cython() ==result
 assertparse_cssutils() ==result
 version='.'.join(map(str, sys.version_info[:3]))
 print('Python{}, consistency OK.'.format(version))
 
 
+defwarm_up():
+is_pypy=hasattr(sys, 'pypy_translation_info')
+ifis_pypy:
+print('Warming up for PyPy...')
+foriinrange(80):
+foriinrange(10):
+parse_python()
+parse_cssutils()
+sys.stdout.write('.')
+sys.stdout.flush()
+sys.stdout.write('\n')
+
+
 deftime(function):
 seconds=timeit.Timer(function).repeat(TIMEIT_REPEAT, TIMEIT_NUMBER)
 miliseconds=int(min(seconds) *1000)
 returnmiliseconds
 
 
 defrun():
-data_set= [
- ('tinycss + speedups ', parse_cython),
+iftokenizer.cython_tokenize_flat:
+data_set= [
+ ('tinycss + speedups ', parse_cython),
+ ]
+else:
+print('Speedups are NOT available.')
+data_set= []
+data_set+= [
  ('tinycss WITHOUT speedups', parse_python),
  ('cssutils ', parse_cssutils),
  ]
@@ -113,4 +133,5 @@ def run():
 
 if__name__=='__main__':
 check_consistency()
+warm_up()
 run()