Pass through all setUp and tearDown arguments

spulec · Aug 1, 2022 · 33fdce8 · 33fdce8
1 parent f9ee0a8
commit 33fdce8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/freezegun/api.py b/freezegun/api.py
@@ -593,14 +593,14 @@ def tearDownClass(cls):
             orig_setUp = klass.setUp
             orig_tearDown = klass.tearDown
 
-            def setUp(true_self):
+            def setUp(*args, **kwargs):
                 self.start()
                 if orig_setUp is not None:
-                    orig_setUp(true_self)
+                    orig_setUp(*args, **kwargs)
 
-            def tearDown(true_self):
+            def tearDown(*args, **kwargs):
                 if orig_tearDown is not None:
-                    orig_tearDown(true_self)
+                    orig_tearDown(*args, **kwargs)
                 self.stop()
 
             klass.setUp = setUp