Format unit tests

zillow · Sep 7, 2024 · 167402a · 167402a
1 parent 02cef9b
commit 167402a
Showing 1 changed file with 8 additions and 16 deletions.
diff --git a/quantile_forest/tests/test_quantile_forest.py b/quantile_forest/tests/test_quantile_forest.py
@@ -567,7 +567,12 @@ def test_predict_quantiles(
     aggregate_leaves_first,
 ):
     check_predict_quantiles(
-        name, max_samples, max_samples_leaf, quantiles, weighted_quantile, aggregate_leaves_first
+        name,
+        max_samples,
+        max_samples_leaf,
+        quantiles,
+        weighted_quantile,
+        aggregate_leaves_first,
     )
 
 
@@ -1539,14 +1544,7 @@ def _dicts_to_input_pairs(input_dicts):
 
     # Check error if invalid parameters.
     assert_raises(TypeError, calc_weighted_quantile, [1, 2], [1, 1], 0.5)
-    assert_raises(
-        TypeError,
-        calc_weighted_quantile,
-        [1, 2],
-        [1, 1],
-        [0.5],
-        interpolation=None,
-    )
+    assert_raises(TypeError, calc_weighted_quantile, [1, 2], [1, 1], [0.5], interpolation=None)
 
 
 def test_calc_quantile_rank():
@@ -1596,10 +1594,4 @@ def test_calc_quantile_rank():
 
     # Check error if invalid parameters.
     assert_raises(TypeError, calc_quantile_rank, [1, 2], [1])
-    assert_raises(
-        TypeError,
-        calc_quantile_rank,
-        [1, 2],
-        float(1),
-        kind=None,
-    )
+    assert_raises(TypeError, calc_quantile_rank, [1, 2], float(1), kind=None)