No longer ignores reset=True when using optimize_context as decorator.

This commit is contained in:
Quantum 2017-08-14 18:30:55 -04:00
parent ab706fb23a
commit c2ff12a3e9
3 changed files with 12 additions and 2 deletions

View file

@ -66,6 +66,6 @@ class optimize_context(with_metaclass(NoArgDecoratorMeta)):
def __call__(self, function):
@wraps(function)
def wrapper(*args, **kwargs):
with optimize_context(self.callbacks):
with optimize_context(self.callbacks, reset=self.reset):
return function(*args, **kwargs)
return wrapper

View file

@ -27,6 +27,16 @@ class OptimizeContextTest(TestCase):
with optimize_context([3]):
self.assertEqual(config.get_callbacks(), [1, 2, 3])
@optimize_context([3], reset=True)
def test():
self.assertEqual(config.get_callbacks(), [3])
test()
@optimize_context([3])
def test():
self.assertEqual(config.get_callbacks(), [1, 2, 3])
test()
self.assertEqual(config.get_callbacks(), [1, 2])
config.register_callback(4)

View file

@ -4,7 +4,7 @@ from setuptools import setup, find_packages
setup(
name='optimize-later',
version='0.1',
version='0.1.1',
description='Mark potentially slow blocks for notifications when it actually turns out too slow, '
'so you can optimize it.',
author='Quantum',