|
@@ -345,10 +345,11 @@ def define_perf_test(target_class, test_function, optimization_level):
|
|
|
'test_%s' % optimization_level.replace('-', '_').lower(),
|
|
|
lambda self: test_function(self, optimization_level))
|
|
|
|
|
|
-def define_perf_tests(target_class, test_function):
|
|
|
+def define_perf_tests(target_class, test_function, optimization_levels=None):
|
|
|
"""Defines performance tests in the given class. Each test calls the given function."""
|
|
|
- for optimization_level in ALL_OPTIMIZATION_LEVELS:
|
|
|
- define_perf_test(target_class, test_function, optimization_level)
|
|
|
+ optimization_levels = ALL_OPTIMIZATION_LEVELS
|
|
|
+ for opt_level in optimization_levels:
|
|
|
+ define_perf_test(target_class, test_function, opt_level)
|
|
|
|
|
|
def get_model_constructor(code):
|
|
|
# First change multiple spaces to a tab
|