fix test cases

ricequant · Dec 4, 2018 · 8e826cb · 8e826cb
1 parent 9f09df5
commit 8e826cb
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/api/test_api_base.py b/tests/api/test_api_base.py
@@ -349,16 +349,18 @@ def handle_bar(context, bar_dict):
     return init, handle_bar
 
 
-@as_test_strategy
+@as_test_strategy()
 def test_subscribe_event():
+    flags = {}
+
     def init(_):
         subscribe_event(EVENT.BEFORE_TRADING, on_before_trading)
 
     def before_trading(_):
-        g.before_trading_ran = True
+        flags["before_trading_ran"] = True
 
     def on_before_trading(_):
-        assert g.before_trading_ran
-        g.before_trading_ran = False
+        assert flags["before_trading_ran"]
+        flags["before_trading_ran"] = False
 
     return init, before_trading