From beea8624d1101a97c31db988dc503addbbe2c305 Mon Sep 17 00:00:00 2001 From: vikram-narayan Date: Fri, 27 Jan 2017 18:19:23 -0500 Subject: [PATCH] MAINT: pass args, kwargs to add_class_callback --- zipline/testing/fixtures.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/zipline/testing/fixtures.py b/zipline/testing/fixtures.py index 3b544078..8dd09d21 100644 --- a/zipline/testing/fixtures.py +++ b/zipline/testing/fixtures.py @@ -135,7 +135,7 @@ class ZiplineTestCase(with_metaclass(FinalMeta, TestCase)): @final @classmethod - def add_class_callback(cls, callback): + def add_class_callback(cls, callback, *args, **kwargs): """ 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.' '\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 def setUp(self):