fix tests

Author: TexasCoding

tests/indicators/conftest.py

@@ -1,5 +1,6 @@
-import pytest
 import polars as pl
+import pytest
+
 
 @pytest.fixture
 def sample_ohlcv_df():
@@ -8,24 +9,29 @@ def sample_ohlcv_df():
     Values are monotonically increasing for easy/deterministic indicator output.
     """
     n = 120
-    return pl.DataFrame({
-        "open":   [float(i) for i in range(n)],
-        "high":   [float(i) + 1 for i in range(n)],
-        "low":    [float(i) - 1 for i in range(n)],
-        "close":  [float(i) + 0.5 for i in range(n)],
-        "volume": [100 + i for i in range(n)],
-    })
+    return pl.DataFrame(
+        {
+            "open": [float(i) for i in range(n)],
+            "high": [float(i) + 1 for i in range(n)],
+            "low": [float(i) - 1 for i in range(n)],
+            "close": [float(i) + 0.5 for i in range(n)],
+            "volume": [100 + i for i in range(n)],
+        }
+    )
+
 
 @pytest.fixture
 def small_ohlcv_df():
     """
     Returns a polars DataFrame with 5 rows to trigger insufficient data paths.
     """
     n = 5
-    return pl.DataFrame({
-        "open":   [float(i) for i in range(n)],
-        "high":   [float(i) + 1 for i in range(n)],
-        "low":    [float(i) - 1 for i in range(n)],
-        "close":  [float(i) + 0.5 for i in range(n)],
-        "volume": [100 + i for i in range(n)],
-    })
+    return pl.DataFrame(
+        {
+            "open": [float(i) for i in range(n)],
+            "high": [float(i) + 1 for i in range(n)],
+            "low": [float(i) - 1 for i in range(n)],
+            "close": [float(i) + 0.5 for i in range(n)],
+            "volume": [100 + i for i in range(n)],
+        }
+    )

tests/indicators/test_all_indicators.py

@@ -1,13 +1,17 @@
-import pytest
-import polars as pl
-import inspect
 import importlib
+import inspect
 import pkgutil
+
+import polars as pl
+import pytest
+
 from project_x_py.indicators.base import BaseIndicator
 
+
 def _concrete_indicator_classes():
     # Recursively discover all non-abstract subclasses of BaseIndicator in project_x_py.indicators.*
     import project_x_py.indicators
+
     seen = set()
     result = []
 
@@ -27,7 +31,9 @@ def onclass(cls):
         result.append(cls)
 
     # Walk all modules in project_x_py.indicators package
-    for finder, name, ispkg in pkgutil.walk_packages(project_x_py.indicators.__path__, project_x_py.indicators.__name__ + "."):
+    for _, name, _ in pkgutil.walk_packages(
+        project_x_py.indicators.__path__, project_x_py.indicators.__name__ + "."
+    ):
         try:
             mod = importlib.import_module(name)
         except Exception:
@@ -37,7 +43,10 @@ def onclass(cls):
     # Remove duplicates, sort by class name for determinism
     return sorted(set(result), key=lambda cls: cls.__name__)
 
-@pytest.mark.parametrize("indicator_cls", _concrete_indicator_classes(), ids=lambda cls: cls.__name__)
+
+@pytest.mark.parametrize(
+    "indicator_cls", _concrete_indicator_classes(), ids=lambda cls: cls.__name__
+)
 def test_indicator_calculate_adds_new_column(indicator_cls, sample_ohlcv_df):
     """
     For every indicator class: instantiate with default ctor, call .calculate() or __call__ on sample data.
@@ -53,17 +62,23 @@ def test_indicator_calculate_adds_new_column(indicator_cls, sample_ohlcv_df):
     except Exception:
         out_df = instance.calculate(sample_ohlcv_df)
 
-    assert isinstance(out_df, pl.DataFrame), f"{indicator_cls.__name__} output is not a polars.DataFrame"
+    assert isinstance(out_df, pl.DataFrame), (
+        f"{indicator_cls.__name__} output is not a polars.DataFrame"
+    )
     assert out_df.height == sample_ohlcv_df.height, (
         f"{indicator_cls.__name__} output row count {out_df.height} != input {sample_ohlcv_df.height}"
     )
     new_cols = set(out_df.columns) - input_cols
     assert new_cols, f"{indicator_cls.__name__} did not add any new columns"
 
+
 def _get_new_column_names(indicator_cls, input_cols, df):
     return set(df.columns) - set(input_cols)
 
-@pytest.mark.parametrize("indicator_cls", _concrete_indicator_classes(), ids=lambda cls: cls.__name__)
+
+@pytest.mark.parametrize(
+    "indicator_cls", _concrete_indicator_classes(), ids=lambda cls: cls.__name__
+)
 def test_indicator_caching_returns_same_object(indicator_cls, sample_ohlcv_df):
     """
     Calling the indicator twice with the same df on the same instance should return the exact same DataFrame object (proves internal cache).
@@ -74,4 +89,4 @@ def test_indicator_caching_returns_same_object(indicator_cls, sample_ohlcv_df):
     out2 = instance(sample_ohlcv_df)
     assert out1 is out2, (
         f"{indicator_cls.__name__} did not return identical object on repeated call (cache broken?)"
-    )
+    )

tests/test_client.py

@@ -1,7 +1,5 @@
 """Main test entry point for ProjectX client module tests."""
 
-import pytest
-
 from project_x_py import ProjectX
 
 # This file serves as a main entry point for running client tests.