From 3073cb225ea2b1f734ffdd6a07f79c2bb1455013 Mon Sep 17 00:00:00 2001 From: AliGheshlaghi Date: Tue, 25 Feb 2025 19:05:03 +0330 Subject: [PATCH] new-indicator: change tests to remove some of assertion errors. --- tests/test_studies.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_studies.py b/tests/test_studies.py index ff234c7e..9fa3f2b4 100644 --- a/tests/test_studies.py +++ b/tests/test_studies.py @@ -9,7 +9,7 @@ [pytest.param(ta.CommonStudy, id="common"), pytest.param(ta.AllStudy, id="all")] # +/- when adding/removing indicators -ALL_COLUMNS = 332 +ALL_COLUMNS = 335 def test_all_study_props(all_study): @@ -33,7 +33,7 @@ def test_common_study_props(common_study): @pytest.mark.parametrize("category,columns", [ ("candles", 70), ("cycles", 2), ("momentum", 85), ("overlap", 56), ("performance", 2), ("statistics", 16), ("transform", 5), ("trend", 33), - ("volatility", 36), ("volume", 27), + ("volatility", 39), ("volume", 27), pytest.param(ta.AllStudy, ALL_COLUMNS, id=f"all-{ALL_COLUMNS}"), pytest.param(ta.CommonStudy, 5, id="common-5"), ]) @@ -89,7 +89,7 @@ def test_study_custom_e_talib(df, custom_study_e, talib): @pytest.mark.parametrize("talib", [False, True]) def test_study_all_multirun_talib(df, all_study, talib): - new_columns = 620 # +/- when adding/removing indicators + new_columns = 623 # +/- when adding/removing indicators initial_columns = df.shape[1] df.ta.study(all_study, length=10, cores=0, talib=talib) df.ta.study(all_study, length=50, cores=0, talib=talib)