From c2ff12a3e97503692672a3dda1d87aaeb00fee79 Mon Sep 17 00:00:00 2001 From: Quantum Date: Mon, 14 Aug 2017 18:30:55 -0400 Subject: [PATCH] No longer ignores reset=True when using optimize_context as decorator. --- optimize_later/config.py | 2 +- optimize_later/tests/test_standard.py | 10 ++++++++++ setup.py | 2 +- 3 files changed, 12 insertions(+), 2 deletions(-) diff --git a/optimize_later/config.py b/optimize_later/config.py index f0cb339..7f574ac 100644 --- a/optimize_later/config.py +++ b/optimize_later/config.py @@ -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 diff --git a/optimize_later/tests/test_standard.py b/optimize_later/tests/test_standard.py index 2510cf2..c9342bf 100644 --- a/optimize_later/tests/test_standard.py +++ b/optimize_later/tests/test_standard.py @@ -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) diff --git a/setup.py b/setup.py index 32e5e86..349f0e4 100644 --- a/setup.py +++ b/setup.py @@ -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',