Skip to content
Snippets Groups Projects
Commit 6cd481c0 authored by Pierre-Yves David's avatar Pierre-Yves David :octopus:
Browse files

benchmark: introduce a 'benchmark_name' decorator

This will make it simple to pick a benchmark name.
parent b03cb8ea
No related branches found
No related tags found
No related merge requests found
......@@ -2,6 +2,7 @@
from .utils import (
BaseTestSuite,
benchmark_name,
params_as_kwargs,
STRIP_VARIANTS_LIST,
)
......@@ -52,6 +53,7 @@
role_action = 'discovery'
role_subtype = 'identical'
@benchmark_name('exchange.discovery.changesets.track_identical')
@params_as_kwargs
def track_identical(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
......@@ -55,7 +57,6 @@
@params_as_kwargs
def track_identical(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
track_identical.benchmark_name = 'exchange.discovery.changesets.track_identical'
@setup_role
class DiscoverySubsetTimeSuite(BaseDiscoveryTimeSuite):
......@@ -69,6 +70,7 @@
role_action = 'discovery'
role_subtype = 'subset'
@benchmark_name('exchange.discovery.changesets.track_subset')
@params_as_kwargs
def track_discovery_subset(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
......@@ -72,7 +74,6 @@
@params_as_kwargs
def track_discovery_subset(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
track_discovery_subset.benchmark_name = 'exchange.discovery.changesets.track_subset'
@setup_role
class DiscoverySupersetTimeSuite(BaseDiscoveryTimeSuite):
......@@ -86,6 +87,7 @@
role_action = 'discovery'
role_subtype = 'superset'
@benchmark_name('exchange.discovery.changesets.track_superset')
@params_as_kwargs
def track_discovery_superset(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
......@@ -89,7 +91,6 @@
@params_as_kwargs
def track_discovery_superset(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
track_discovery_superset.benchmark_name = 'exchange.discovery.changesets.track_superset'
@setup_role
class DiscoveryBalancedTimeSuite(BaseDiscoveryTimeSuite):
......@@ -104,6 +105,7 @@
role_action = 'discovery'
role_subtype = 'balanced'
@benchmark_name('exchange.discovery.changesets.track_balanced')
@params_as_kwargs
def track_discovery_balanced(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
......@@ -107,7 +109,6 @@
@params_as_kwargs
def track_discovery_balanced(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
track_discovery_balanced.benchmark_name = 'exchange.discovery.changesets.track_balanced'
@setup_role
......@@ -122,6 +123,7 @@
role_action = 'discovery'
role_subtype = 'large-subset'
@benchmark_name('exchange.discovery.changesets.track_large_subset')
@params_as_kwargs
def track_discovery_large_subset(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
......@@ -125,7 +127,6 @@
@params_as_kwargs
def track_discovery_large_subset(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
track_discovery_large_subset.benchmark_name = 'exchange.discovery.changesets.track_large_subset'
@setup_role
class DiscoveryLargeSupersetTimeSuite(BaseDiscoveryTimeSuite):
......@@ -139,6 +140,7 @@
role_action = 'discovery'
role_subtype = 'large-superset'
@benchmark_name('exchange.discovery.changesets.track_large_superset')
@params_as_kwargs
def track_discovery_large_superset(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
......@@ -142,7 +144,6 @@
@params_as_kwargs
def track_discovery_large_superset(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
track_discovery_large_superset.benchmark_name = 'exchange.discovery.changesets.track_large_superset'
@setup_role
class DiscoveryLargeBalancedTimeSuite(BaseDiscoveryTimeSuite):
......@@ -157,6 +158,7 @@
role_action = 'discovery'
role_subtype = 'large-balanced'
@benchmark_name('exchange.discovery.changesets.track_large_balanced')
@params_as_kwargs
def track_discovery_large_balanced(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
......@@ -160,4 +162,3 @@
@params_as_kwargs
def track_discovery_large_balanced(self, *args, **kwargs):
return self._track_discovery(self, *args, **kwargs)
track_discovery_large_balanced.benchmark_name = 'exchange.discovery.changesets.track_large_balanced'
......@@ -286,7 +286,12 @@
return decorator
def benchmark_name(name):
"""decorate a benchmark method to set its benchmark name"""
def dec(method):
method.benchmark_name = name
return method
return dec
class BaseTestSuite(object):
timer = timeit.default_timer
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment