mirror of
https://github.com/wassname/catalyst.git
synced 2026-08-17 11:16:40 +08:00
Merge pull request #1666 from quantopian/add_class_callback_args
MAINT: pass args, kwargs to add_class_callback
This commit is contained in:
@@ -135,7 +135,7 @@ class ZiplineTestCase(with_metaclass(FinalMeta, TestCase)):
|
|||||||
|
|
||||||
@final
|
@final
|
||||||
@classmethod
|
@classmethod
|
||||||
def add_class_callback(cls, callback):
|
def add_class_callback(cls, callback, *args, **kwargs):
|
||||||
"""
|
"""
|
||||||
Register a callback to be executed during tearDownClass.
|
Register a callback to be executed during tearDownClass.
|
||||||
|
|
||||||
@@ -149,7 +149,7 @@ class ZiplineTestCase(with_metaclass(FinalMeta, TestCase)):
|
|||||||
'Attempted to add a class callback in init_instance_fixtures.'
|
'Attempted to add a class callback in init_instance_fixtures.'
|
||||||
'\nDid you mean to call add_instance_callback?',
|
'\nDid you mean to call add_instance_callback?',
|
||||||
)
|
)
|
||||||
return cls._class_teardown_stack.callback(callback)
|
return cls._class_teardown_stack.callback(callback, *args, **kwargs)
|
||||||
|
|
||||||
@final
|
@final
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
|
|||||||
Reference in New Issue
Block a user