From 447f3c860dba356d949aeec9b4a4047bebc0cf77 Mon Sep 17 00:00:00 2001 From: Lukas Weidenholzer Date: Fri, 18 Aug 2023 11:58:37 +0200 Subject: [PATCH] add assert to test --- tests/test_arrays.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/test_arrays.py b/tests/test_arrays.py index 42f642bc..e052547f 100644 --- a/tests/test_arrays.py +++ b/tests/test_arrays.py @@ -161,7 +161,12 @@ def test_array_contains(data, value, expected): def test_array_contains_axis(): data = np.array([[4, 5, 6], [5, 7, 9]]) - result = array_contains(data, 5, axis=1) + + result_0 = array_contains(data, 5, axis=0) + np.testing.assert_array_equal(result_0, np.array([True, True, False])) + + result_1 = array_contains(data, 5, axis=1) + np.testing.assert_array_equal(result_1, np.array([True, True])) def test_array_contains_object_dtype():