|
10 | 10 |
|
11 | 11 | import numpy as np
|
12 | 12 |
|
13 |
| -from sklearn.utils.testing import assert_almost_equal |
14 | 13 | from sklearn.utils.testing import assert_array_equal
|
15 | 14 | from sklearn.utils.testing import assert_array_almost_equal
|
16 | 15 | from sklearn.utils.testing import assert_raises
|
@@ -268,10 +267,17 @@ def test_iforest_average_path_length():
|
268 | 267 | result_two = 2. * (np.log(998.) + np.euler_gamma) - 2. * 998. / 999.
|
269 | 268 |
|
270 | 269 | assert_array_almost_equal(_average_path_length([1]), [1.], decimal=10)
|
271 |
| - assert_array_almost_equal(_average_path_length([5]), [result_one], decimal=10) |
272 |
| - assert_array_almost_equal(_average_path_length([999]), [result_two], decimal=10) |
273 |
| - assert_array_almost_equal(_average_path_length(np.array([1, 5, 999])), |
274 |
| - [1., result_one, result_two], decimal=10) |
| 270 | + assert_array_almost_equal( |
| 271 | + _average_path_length([5]), [result_one], decimal=10 |
| 272 | + ) |
| 273 | + assert_array_almost_equal( |
| 274 | + _average_path_length([999]), [result_two], decimal=10 |
| 275 | + ) |
| 276 | + assert_array_almost_equal( |
| 277 | + _average_path_length(np.array([1, 5, 999])), |
| 278 | + [1.0, result_one, result_two], |
| 279 | + decimal=10, |
| 280 | + ) |
275 | 281 |
|
276 | 282 |
|
277 | 283 | @pytest.mark.filterwarnings('ignore:default contamination')
|
|
0 commit comments