Skip to content

Commit

Permalink
new-indicator: change tests to remove some of assertion errors.
Browse files Browse the repository at this point in the history
  • Loading branch information
aligheshlaghi97 committed Feb 25, 2025
1 parent d7efc59 commit 3073cb2
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/test_studies.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand All @@ -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"),
])
Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit 3073cb2

Please sign in to comment.