From 08c1ece098ee23792e3d0d36ad607c5fc89beebc Mon Sep 17 00:00:00 2001 From: Lorenzo Stella Date: Thu, 17 Oct 2024 11:45:26 +0200 Subject: [PATCH] np.alltrue -> np.all --- test/transform/test_transform.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/transform/test_transform.py b/test/transform/test_transform.py index fbcdfb9d16..bc9815427c 100644 --- a/test/transform/test_transform.py +++ b/test/transform/test_transform.py @@ -155,8 +155,8 @@ def test_AddTimeFeatures(start, target, is_train: bool): tmp_idx = pd.period_range( start=start, freq=start.freq, periods=expected_length ) - assert np.alltrue(mat[0] == time_feature.day_of_week(tmp_idx)) - assert np.alltrue(mat[1] == time_feature.day_of_month(tmp_idx)) + assert np.all(mat[0] == time_feature.day_of_week(tmp_idx)) + assert np.all(mat[1] == time_feature.day_of_month(tmp_idx)) @pytest.mark.parametrize("is_train", TEST_VALUES["is_train"]) @@ -285,7 +285,7 @@ def test_InstanceSplitter( # expected_length = len(target) + (0 if is_train else pred_length) # assert len(out['age']) == expected_length - # assert np.alltrue(out['age'] == np.log10(2.0 + np.arange(expected_length))) + # assert np.all(out['age'] == np.log10(2.0 + np.arange(expected_length))) @pytest.mark.parametrize("is_train", TEST_VALUES["is_train"])