From 6c25feabf26ccdd240dca04fc422f1b06feeedf6 Mon Sep 17 00:00:00 2001 From: Matthias Date: Fri, 6 Dec 2024 20:04:59 +0100 Subject: [PATCH] tests: assert type of orderflow object --- tests/data/test_converter_orderflow.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/data/test_converter_orderflow.py b/tests/data/test_converter_orderflow.py index ded6d2088..3bf7faf58 100644 --- a/tests/data/test_converter_orderflow.py +++ b/tests/data/test_converter_orderflow.py @@ -551,6 +551,9 @@ def test_analyze_with_orderflow( assert isinstance(lastval_trades, list) assert len(lastval_trades) == 122 + lastval_of = df1.at[len(df1) - 1, "orderflow"] + assert isinstance(lastval_of, dict) + spy.reset_mock() # Ensure caching works - call the same logic again. df2 = strategy.advise_indicators(ohlcv_history, {"pair": pair}) @@ -564,3 +567,10 @@ def test_analyze_with_orderflow( assert ( df2[col].count() == 5 ), f"Round2: Column {col} has {df2[col].count()} non-NaN values" + + lastval_trade2 = df2.at[len(df2) - 1, "trades"] + assert isinstance(lastval_trade2, list) + assert len(lastval_trade2) == 122 + + lastval_of2 = df2.at[len(df2) - 1, "orderflow"] + assert isinstance(lastval_of2, dict)