update test case for commit 134aea

ricequant · Jul 20, 2018 · 23e0bd9 · 23e0bd9
1 parent 134aea1
commit 23e0bd9
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/tests/mod/sys_accounts/test_account_model.py b/tests/mod/sys_accounts/test_account_model.py
@@ -10,7 +10,7 @@
             "end_date": "2016-03-08",
             "frequency": "1d",
             "accounts": {
-                "stock": 100000000
+                "stock": 1000000,
             }
         },
         "extra": {
@@ -20,27 +20,33 @@
             "sys_progress": {
                 "enabled": True,
                 "show": True,
-            },
+            }
         },
     }, test_strategies)
 
 
 @as_test_strategy({
     "base": {
-        "start_date": "2015-12-25",
+        "start_date": "2015-12-07",
         "end_date": "2016-01-05"
     }
 })
 def test_stock_account_settlement():
+    import datetime
+
     def init(context):
         # 招商地产
         context.s = "000024.XSHE"
-        context.counter = 0
+        context.fired = False
+        context.total_value_before_delisted = None
 
     def handle_bar(context, _):
-        context.counter += 1
-        if context.counter == 1:
-            order_shares(context.s, 10000)
-        print(context.portfolio.total_value)
+        if not context.fired:
+            order_shares(context.s, 20000)
+            context.fired = True
+        if context.now.date() == datetime.date(2015, 12, 29):
+            context.total_value_before_delisted = context.portfolio.total_value
+        if context.now.date() > datetime.date(2015, 12, 29):
+            assert context.portfolio.total_value == context.total_value_before_delisted
 
     return init, handle_bar