0.5.1: 新增测试

test/test_analyze.py

@@ -6,14 +6,32 @@ from cobra.data.kline import get_kline
 sys.path.insert(0, '.')
 sys.path.insert(0, '..')
 import czsc
-from czsc.analyze import KlineAnalyze
+from czsc import KlineAnalyze, is_bei_chi
 
 warnings.warn(f"czsc version is {czsc.__version__}")
 
-df = get_kline(ts_code="000001.SH", end_dt="2020-07-15 15:00:00", freq='D', asset='I')
-split_dt = "2020-07-01 15:00:00"
-ka = KlineAnalyze(df[df.dt <= split_dt], name="日线")
+df = get_kline(ts_code="000001.SH", end_dt="2020-07-16 15:00:00", freq='D', asset='I')
+ka = KlineAnalyze(df, name="日线")
 print(ka)
-bars = df[df.dt > split_dt].to_dict("records")
+
+
+def test_kline_analyze():
+    assert ka.bis[-1].mark == 'g'
+    assert ka.xds[-2].mark == 'd'
+
+    ka.to_html("kline.html")
+    ka.to_image("kline.png")
+
+
+def test_bei_chi():
+    # 线段背驰
+    zs1 = {"start_dt": ka.xds[-3].dt, "end_dt": ka.xds[-2].dt, "direction": "down"}
+    zs2 = {"start_dt": ka.xds[-5].dt, "end_dt": ka.xds[-4].dt, "direction": "down"}
+    assert is_bei_chi(ka, zs1, zs2, mode='xd', adjust=0.9)
+
+    # 笔背驰
+    zs1 = {"start_dt": '2019-05-17 15:00:00', "end_dt": '2019-05-27 15:00:00'}
+    zs2 = {"start_dt": '2019-04-08 15:00:00', "end_dt": '2019-05-10 15:00:00'}
+    assert is_bei_chi(ka, zs1, zs2, mode='bi', adjust=0.9)