DOC: require pyarrow (#151)

* DOC: require pyarrow * Update tests to work with pandas 2.2.0
audeering · Jan 24, 2024 · 28a2f59 · 28a2f59
1 parent f19782e
commit 28a2f59
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -4,6 +4,7 @@ auditok
 ipykernel
 jupyter-sphinx
 librosa
+pyarrow
 sphinx
 sphinx-audeering-theme >=1.2.1
 sphinx-autodoc-typehints

diff --git a/tests/test_process.py b/tests/test_process.py
@@ -757,7 +757,7 @@ def test_process_index_filewise_end_times(tmpdir):
             None,
             None,
             False,
-            np.array([1., 2., 3.]),
+            np.array([[1., 2., 3.]]),
         ),
         (
             None,
@@ -769,7 +769,7 @@ def test_process_index_filewise_end_times(tmpdir):
             None,
             None,
             False,
-            np.array([1., 2., 3.]),
+            np.array([[1., 2., 3.]]),
         ),
         (
             None,
@@ -781,7 +781,7 @@ def test_process_index_filewise_end_times(tmpdir):
             None,
             None,
             False,
-            np.array([1., 2.]),
+            np.array([[1., 2.]]),
         ),
         (
             None,
@@ -793,7 +793,7 @@ def test_process_index_filewise_end_times(tmpdir):
             None,
             None,
             False,
-            np.array([1., 2.]),
+            np.array([[1., 2.]]),
         ),
         (
             signal_max,
@@ -993,7 +993,6 @@ def test_process_signal(
         start=start,
         end=end,
     )
-    signal = np.atleast_2d(signal)
     if start is None or pd.isna(start):
         start = pd.to_timedelta(0)
     elif isinstance(start, (int, float)):
@@ -1002,7 +1001,7 @@ def test_process_signal(
         start = pd.to_timedelta(start)
     if end is None or (pd.isna(end) and not keep_nat):
         end = pd.to_timedelta(
-            signal.shape[1] / sampling_rate,
+            np.atleast_2d(signal).shape[1] / sampling_rate,
             unit='s',
         )
     elif isinstance(end, (int, float)):