From 4cb9f55ca907a0b48f392e65c6cb5f5eb3f9c49e Mon Sep 17 00:00:00 2001 From: Neyberson Date: Thu, 6 Jun 2024 15:44:40 -0500 Subject: [PATCH] Improve asserts from test_append function at test_array --- tests/utils/test_array.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/tests/utils/test_array.py b/tests/utils/test_array.py index ab49e67..f79f92c 100644 --- a/tests/utils/test_array.py +++ b/tests/utils/test_array.py @@ -183,13 +183,13 @@ def test_round(): def test_append(): - assert pzutils.append(np_data, 1) is not None - assert pzutils.append(df_pandas_irregular["depth"], [2,3,4]) is not None - assert pzutils.append(series, (2,3,4,5,6,7)) is not None - assert pzutils.append(series_polars, 8) is not None - assert pzutils.append(list_data, np.array([1,2,3,4])) is not None - assert pzutils.append(np_data_irregular, (2,3,4,5,6,7)) is not None - assert pzutils.append(df_polars_irregular["depth"], np.array([1,2,3,4])) is not None - assert pzutils.append(series_data_irregular, np.array([1,2,3,4])) is not None - assert pzutils.append(series_polars_data_irregular, (2,3,4,5,6,7)) is not None - assert pzutils.append(list_data_irregular, 1000) is not None + assert (pzutils.append(np_data, 1))[-1] == 1 + assert (pzutils.append(df_pandas_irregular["depth"], [2,3,4])).iloc[-1] == 4 + assert (pzutils.append(series, (2,3,4,5,6,7))).iloc[-1] == 7 + assert (pzutils.append(series_polars, 8))[-1] == 8 + assert (pzutils.append(list_data, np.array([1,2,3,4])))[-1] == 4 + assert (pzutils.append(np_data_irregular, (2,3,4,5,6,7)))[-1] == 7 + assert (pzutils.append(df_polars_irregular["depth"], np.array([1,2,3,4])))[-1] == 4 + assert (pzutils.append(series_data_irregular, np.array([1,2,3,4]))).iloc[-1] == 4 + assert (pzutils.append(series_polars_data_irregular, (2,3,4,5,6,7)))[-1] == 7 + assert (pzutils.append(list_data_irregular, 1000))[-1] == 1000