sanguo/sanguo_quant_live
8
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

auto-sync: 2026-04-17 20:24:35

Browse Source
This commit is contained in:
cfdaily
2026-04-17 20:24:35 +08:00
parent 9ec297401a
commit 8cf2914bfa
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zhangfei-technical/research/task-20260417-atr-indicator/test_atr.py
+2 -1
View File
@@ -167,7 +167,8 @@ def test_calculate_atr_convenience():
assert 'TR' in result.columns assert 'TR' in result.columns
assert 'ATR_5' in result.columns assert 'ATR_5' in result.columns
assert result.shape == df.shape assert result.shape[0] == df.shape[0]
assert result.shape[1] == df.shape[1] + 2
result_dropna = calculate_atr(df, period=5, method='sma', drop_na=True) result_dropna = calculate_atr(df, period=5, method='sma', drop_na=True)
assert len(result_dropna) < len(df) assert len(result_dropna) < len(df)