def test_find_zs():
assert ka.down_zs_number() == 2
assert ka.up_zs_number() == 1
xd_zs = find_zs(ka.xd)
bi_zs = find_zs(ka.bi)
assert xd_zs[-2]["ZD"] == 2850.71
assert xd_zs[-2]["ZG"] == 3684.57
assert xd_zs[-1]["ZD"] == 2691.02
assert xd_zs[-1]["ZG"] == 2827.34
assert bi_zs[-2]['ZD'] == 2987.77
assert bi_zs[-2]['ZG'] == 3125.02
assert bi_zs[-1]['ZD'] == 2838.38
assert bi_zs[-1]['ZG'] == 2956.78
def test_find_zs():
# 造数测试
points = [
{"dt": 0, "fx_mark": "d", "xd": 8},
{"dt": 1, "fx_mark": "g", "xd": 10},
{"dt": 2, "fx_mark": "d", "xd": 9},
{"dt": 3, "fx_mark": "g", "xd": 11},
{"dt": 4, "fx_mark": "d", "xd": 10.5},
{"dt": 5, "fx_mark": "g", "xd": 12},
{"dt": 6, "fx_mark": "d", "xd": 11.1},
{"dt": 7, "fx_mark": "g", "xd": 14},
{"dt": 8, "fx_mark": "d", "xd": 13},
{"dt": 9, "fx_mark": "g", "xd": 13.8},
{"dt": 10, "fx_mark": "d", "xd": 12.9},
{"dt": 11, "fx_mark": "g", "xd": 14.5},
{"dt": 12, "fx_mark": "d", "xd": 13.2},
{"dt": 13, "fx_mark": "g", "xd": 15},
{"dt": 14, "fx_mark": "d", "xd": 14.3},
{"dt": 15, "fx_mark": "g", "xd": 16.2},
{"dt": 16, "fx_mark": "d", "xd": 15.3},
{"dt": 17, "fx_mark": "g", "xd": 17.6},
{"dt": 18, "fx_mark": "d", "xd": 15.9},
{"dt": 19, "fx_mark": "g", "xd": 18.2},
{"dt": 20, "fx_mark": "d", "xd": 16.8},
{"dt": 21, "fx_mark": "g", "xd": 17.8},
{"dt": 22, "fx_mark": "d", "xd": 16.9},
{"dt": 23, "fx_mark": "g", "xd": 18.1},
]
zss = find_zs(points[:8])
assert len(zss) == 1
zss = find_zs(points[:15])
assert len(zss) == 2
zss = find_zs(points)
assert len(zss) == 3 and zss[0]['ZG'] < zss[1]['ZD'] and zss[1]['ZG'] < zss[2]['ZD']
# 获取用于比较趋势背驰的两端
fd1 = [x for x in points if x['dt'] >= zss[2]['end_point']['dt']]
fd2 = [x for x in points if zss[2]['start_point']['dt'] > x['dt'] >= zss[1]['end_point']['dt']]
fd3 = [x for x in points if zss[1]['start_point']['dt'] > x['dt'] >= zss[0]['end_point']['dt']]
fd4 = [x for x in points if x['dt'] <= zss[0]['start_point']['dt']]
assert fd1[0]['fx_mark'] == fd2[0]['fx_mark'] == fd3[0]['fx_mark'] == fd4[0]['fx_mark'] == 'd'
assert fd1[-1]['fx_mark'] == fd2[-1]['fx_mark'] == fd3[-1]['fx_mark'] == fd4[-1]['fx_mark'] == 'g'
def test_find_zs():
bi_zs = find_zs(ka.bi_list)
xd_zs = find_zs(ka.xd_list)