refactor: improve assertion clarity and numerical tolerance in tests

- Updated assertion formatting in integration tests for better readability and consistency in error messages.
- Adjusted numerical tolerance in `test_pdex.py` to allow for smaller differences, enhancing test robustness.

Author: nick-youngblut

tests/test_integration.py

@@ -252,9 +252,9 @@ def _assert_high_correlation(
             # Show sample of mismatches
             print(results.loc[valid_mask, [col_std, col_low]].head())
 
-        assert (
-            corr > threshold
-        ), f"Correlation for {col} is {corr:.6f}, expected > {threshold}"
+        assert corr > threshold, (
+            f"Correlation for {col} is {corr:.6f}, expected > {threshold}"
+        )
 
 
 def test_integration_dense_counts():
@@ -499,6 +499,6 @@ def test_integration_varying_chunk_sizes():
         y = merged.loc[valid_mask, col_small]
 
         # Use allclose for strict equality check since it's the same algorithm
-        assert np.allclose(
-            x, y, equal_nan=True
-        ), f"Mismatch in {col} between chunk sizes"
+        assert np.allclose(x, y, equal_nan=True), (
+            f"Mismatch in {col} between chunk sizes"
+        )

tests/test_pdex.py

@@ -286,7 +286,7 @@ def test_low_memory_float_data_uses_sorting_kernel():
         chunked["p_value"],
         check_exact=False,
         rtol=0,
-        atol=5e-2,
+        atol=5e-3,
     )
     pd.testing.assert_series_equal(
         baseline["statistic"],