diff --git a/test/dynamics/arraylias/test_alias.py b/test/dynamics/arraylias/test_alias.py index b9cb16f92..b2a9f7a6a 100644 --- a/test/dynamics/arraylias/test_alias.py +++ b/test/dynamics/arraylias/test_alias.py @@ -41,7 +41,7 @@ pass -@partial(test_array_backends, array_libraries=["numpy", "jax", "array_numpy", "array_jax"]) +@test_array_backends class TestDynamicsNumpy: """Test cases for global numpy configuration.""" diff --git a/test/dynamics/signals/test_signals.py b/test/dynamics/signals/test_signals.py index 8e0a591ba..82ee86373 100644 --- a/test/dynamics/signals/test_signals.py +++ b/test/dynamics/signals/test_signals.py @@ -34,7 +34,7 @@ # pylint: disable=no-member -@partial(test_array_backends, array_libraries=["numpy", "jax", "array_numpy", "array_jax"]) +@test_array_backends class TestSignal: """Tests for Signal object.""" @@ -313,7 +313,7 @@ def test_conjugate(self): ) -@partial(test_array_backends, array_libraries=["numpy", "jax", "array_numpy", "array_jax"]) +@test_array_backends class TestConstant: """Tests for constant signal object.""" @@ -385,7 +385,7 @@ def test_conjugate(self): self.assertAllClose(const_conj(1.1), 3.0) -@partial(test_array_backends, array_libraries=["numpy", "jax", "array_numpy", "array_jax"]) +@test_array_backends class TestDiscreteSignal: """Tests for DiscreteSignal object.""" @@ -771,13 +771,11 @@ def test_empty_DiscreteSignal_to_sum(self): self.assertTrue(empty_sum.samples.shape == (1, 0)) -test_array_backends(TestSignalSum, array_libraries=["numpy", "jax", "array_numpy", "array_jax"]) -test_array_backends( - TestDiscreteSignalSum, array_libraries=["numpy", "jax", "array_numpy", "array_jax"] -) +test_array_backends(TestSignalSum) +test_array_backends(TestDiscreteSignalSum) -@partial(test_array_backends, array_libraries=["numpy", "jax", "array_numpy", "array_jax"]) +@test_array_backends class TestSignalList: """Test cases for SignalList class.""" @@ -840,7 +838,7 @@ def test_construction_with_numbers(self): self.assertAllClose(sig_list(3.0), self.asarray([4.0, 2.0, 3.0])) -@partial(test_array_backends, array_libraries=["numpy", "jax", "array_numpy", "array_jax"]) +@test_array_backends class TestSignalCollection: """Test cases for SignalCollection functionality."""