add tests

pyfar · Oct 9, 2023 · fe4cf05 · fe4cf05
1 parent c1289b4
commit fe4cf05
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 0 deletions.
diff --git a/tests/test_audio_frequency_data.py b/tests/test_audio_frequency_data.py
@@ -108,6 +108,18 @@ def test_reshape_exceptions():
         data_out = data_in.reshape((3, 4))
 
 
+def test_transpose():
+    signal_in = FrequencyData(np.random.rand(6, 2, 5, 256), range(256))
+    npt.assert_allclose(signal_in.transpose()._data, signal_in.T._data)
+    signal_out = signal_in.transpose()
+    npt.assert_allclose(signal_in._data.transpose(2, 1, 0, 3), signal_out._data)
+    taxis = (2, 0, 1)
+    signal_out = signal_in.transpose(taxis)
+    npt.assert_allclose(signal_in._data.transpose(*taxis, 3), signal_out._data)
+    signal_out = signal_in.transpose(*taxis)
+    npt.assert_allclose(signal_in._data.transpose(*taxis, 3), signal_out._data)
+
+
 def test_flatten():
 
     # test 2D signal (flatten should not change anything)

diff --git a/tests/test_audio_signal.py b/tests/test_audio_signal.py
@@ -423,6 +423,18 @@ def test_reshape_exceptions():
         signal_out = signal_in.reshape((3, 4))
 
 
+def test_transpose():
+    signal_in = Signal(np.random.rand(6, 2, 5, 256), 44100)
+    npt.assert_allclose(signal_in.transpose()._data, signal_in.T._data)
+    signal_out = signal_in.transpose()
+    npt.assert_allclose(signal_in._data.transpose(2, 1, 0, 3), signal_out._data)
+    taxis = (2, 0, 1)
+    signal_out = signal_in.transpose(taxis)
+    npt.assert_allclose(signal_in._data.transpose(*taxis, 3), signal_out._data)
+    signal_out = signal_in.transpose(*taxis)
+    npt.assert_allclose(signal_in._data.transpose(*taxis, 3), signal_out._data)
+
+
 def test_flatten():
 
     # test 2D signal (flatten should not change anything)

diff --git a/tests/test_audio_time_data.py b/tests/test_audio_time_data.py
@@ -92,6 +92,18 @@ def test_reshape_exceptions():
         data_out = data_in.reshape((3, 4))
 
 
+def test_transpose():
+    signal_in = TimeData(np.random.rand(6, 2, 5, 256), range(256))
+    npt.assert_allclose(signal_in.transpose()._data, signal_in.T._data)
+    signal_out = signal_in.transpose()
+    npt.assert_allclose(signal_in._data.transpose(2, 1, 0, 3), signal_out._data)
+    taxis = (2, 0, 1)
+    signal_out = signal_in.transpose(taxis)
+    npt.assert_allclose(signal_in._data.transpose(*taxis, 3), signal_out._data)
+    signal_out = signal_in.transpose(*taxis)
+    npt.assert_allclose(signal_in._data.transpose(*taxis, 3), signal_out._data)
+
+
 def test_flatten():
 
     # test 2D signal (flatten should not change anything)