From 72a819f7f957081d8e9959054c13df97aa97cbee Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 12:28:35 +0200 Subject: [PATCH 01/17] Update test_noises.py --- deeptrack/tests/test_noises.py | 23 ++++++++--------------- 1 file changed, 8 insertions(+), 15 deletions(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index 52723be7e..9af509503 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -30,20 +30,20 @@ def test_Offset(self): def test_Background(self): # Test with DeepTrack Image noise = noises.Background(offset=0.5) - input_image = Image(np.zeros((256, 256))) + input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, np.ndarray) + self.assertIsInstance(output_image, xp.ndarray) self.assertEqual(output_image.shape, (256, 256)) - self.assertTrue(np.all(np.array(output_image) == 0.5)) + self.assertTrue(xp.all(xp.array(output_image) == 0.5)) # Test with NumPy array noise = noises.Background(offset=0.5) - input_image = np.ones((10, 10)) + input_image = xp.ones((10, 10)) output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, np.ndarray) + self.assertIsInstance(output_image, xp.ndarray) self.assertEqual(output_image.shape, (10, 10)) - self.assertTrue(np.all(np.array(output_image) == 1.5)) + self.assertTrue(xp.all(xp.array(output_image) == 1.5)) def test_Gaussian(self): noise = noises.Gaussian(mu=0.1, sigma=0.05) @@ -71,15 +71,8 @@ def test_Poisson(self): @unittest.skipUnless(TORCH_AVAILABLE, "PyTorch is not installed.") class TestNoises_Torch(TestNoises_Numpy): BACKEND = "torch" - - def test_Backgroud(self): - noise = noises.Background(offset=0.25) - input_image = torch.zeros(5,5) - output_image = noise.resolve(input_image) - - self.assertIsInstance(output_image, torch.Tensor) - self.assertEqual(output_image.shape, (5,5)) - self.assertTrue(torch.all(output_image == 0.25).item()) + pass + def test_Gaussian(self): noise = noises.Gaussian(mu=0.1, sigma=0.05) From b777a7ddbca9cf9c1e4474fad841d0e86010e91f Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 12:30:52 +0200 Subject: [PATCH 02/17] imports --- deeptrack/tests/test_noises.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index 9af509503..a33fa0bd8 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -4,12 +4,13 @@ import unittest +import array_api_compat as apc import numpy as np from deeptrack.image import Image from deeptrack import noises -from deeptrack.backend import TORCH_AVAILABLE +from deeptrack.backend import OPENCV_AVAILABLE, TORCH_AVAILABLE, xp from deeptrack.tests import BackendTestBase if TORCH_AVAILABLE: From 85be2794896d48141d13e4917eceb07def64fac8 Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 12:33:54 +0200 Subject: [PATCH 03/17] Update test_noises.py --- deeptrack/tests/test_noises.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index a33fa0bd8..3cb72bc3c 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -34,7 +34,7 @@ def test_Background(self): input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, xp.ndarray) + self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) self.assertTrue(xp.all(xp.array(output_image) == 0.5)) From 197ae406941e1bd63859bc196e61ef75befd0e55 Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 12:37:04 +0200 Subject: [PATCH 04/17] u --- deeptrack/tests/test_noises.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index 3cb72bc3c..206e59ff9 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -34,15 +34,16 @@ def test_Background(self): input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, np.ndarray) + self.assertIsInstance(type(output_image), type(input_image)) self.assertEqual(output_image.shape, (256, 256)) self.assertTrue(xp.all(xp.array(output_image) == 0.5)) - # Test with NumPy array + # Test with arrays noise = noises.Background(offset=0.5) input_image = xp.ones((10, 10)) output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, xp.ndarray) + + self.assertIsInstance(type(output_image), type(input_image)) self.assertEqual(output_image.shape, (10, 10)) self.assertTrue(xp.all(xp.array(output_image) == 1.5)) From 16e52ecc36d4b3927c20b44882ae5a539157a5a1 Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 12:39:02 +0200 Subject: [PATCH 05/17] u --- deeptrack/tests/test_noises.py | 1 - 1 file changed, 1 deletion(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index 206e59ff9..802a76ccd 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -34,7 +34,6 @@ def test_Background(self): input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - self.assertIsInstance(type(output_image), type(input_image)) self.assertEqual(output_image.shape, (256, 256)) self.assertTrue(xp.all(xp.array(output_image) == 0.5)) From 206c3c86237232922cfc09047859139b562e9033 Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 12:41:54 +0200 Subject: [PATCH 06/17] u --- deeptrack/tests/test_noises.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index 802a76ccd..2c592dab3 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -33,7 +33,8 @@ def test_Background(self): noise = noises.Background(offset=0.5) input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - + + #self.assertIsInstance(output_image, input_image) self.assertEqual(output_image.shape, (256, 256)) self.assertTrue(xp.all(xp.array(output_image) == 0.5)) @@ -42,7 +43,7 @@ def test_Background(self): input_image = xp.ones((10, 10)) output_image = noise.resolve(input_image) - self.assertIsInstance(type(output_image), type(input_image)) + #self.assertIsInstance(output_image, input_image) self.assertEqual(output_image.shape, (10, 10)) self.assertTrue(xp.all(xp.array(output_image) == 1.5)) From be76e34b7cbf4ba24bc80f6aae1cf2cd5f937aa5 Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 12:44:04 +0200 Subject: [PATCH 07/17] xp.asarray --- deeptrack/tests/test_noises.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index 2c592dab3..c8076f3cf 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -36,7 +36,7 @@ def test_Background(self): #self.assertIsInstance(output_image, input_image) self.assertEqual(output_image.shape, (256, 256)) - self.assertTrue(xp.all(xp.array(output_image) == 0.5)) + self.assertTrue(xp.all(xp.asarray(output_image) == 0.5)) # Test with arrays noise = noises.Background(offset=0.5) @@ -45,7 +45,7 @@ def test_Background(self): #self.assertIsInstance(output_image, input_image) self.assertEqual(output_image.shape, (10, 10)) - self.assertTrue(xp.all(xp.array(output_image) == 1.5)) + self.assertTrue(xp.all(xp.asarray(output_image) == 1.5)) def test_Gaussian(self): noise = noises.Gaussian(mu=0.1, sigma=0.05) From 7e6cd818775d3d1c090e1d7db93e07a2ffb9783c Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 13:04:30 +0200 Subject: [PATCH 08/17] rest of the functions --- deeptrack/tests/test_noises.py | 54 +++++++++++----------------------- 1 file changed, 17 insertions(+), 37 deletions(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index c8076f3cf..e6049276f 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -18,22 +18,22 @@ class TestNoises_Numpy(BackendTestBase): BACKEND = "numpy" - + def test_Offset(self): noise = noises.Offset(offset=0.5) - input_image = Image(np.zeros((256, 256))) + input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, np.ndarray) + #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) - self.assertTrue(np.all(np.array(output_image) == 0.5)) + self.assertTrue(xp.all(xp.asarray(output_image) == 0.5)) def test_Background(self): # Test with DeepTrack Image noise = noises.Background(offset=0.5) input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - + #self.assertIsInstance(output_image, input_image) self.assertEqual(output_image.shape, (256, 256)) self.assertTrue(xp.all(xp.asarray(output_image) == 0.5)) @@ -42,62 +42,42 @@ def test_Background(self): noise = noises.Background(offset=0.5) input_image = xp.ones((10, 10)) output_image = noise.resolve(input_image) - + #self.assertIsInstance(output_image, input_image) self.assertEqual(output_image.shape, (10, 10)) self.assertTrue(xp.all(xp.asarray(output_image) == 1.5)) def test_Gaussian(self): noise = noises.Gaussian(mu=0.1, sigma=0.05) - input_image = Image(np.zeros((256, 256))) + input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, np.ndarray) + + #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) def test_ComplexGaussian(self): noise = noises.ComplexGaussian(mu=0.1, sigma=0.05) - input_image = Image(np.zeros((256, 256))) + input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, np.ndarray) + + #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) - self.assertTrue(np.iscomplexobj(output_image)) + self.assertTrue(xp.iscomplexobj(output_image)) def test_Poisson(self): noise = noises.Poisson(snr=20) - input_image = Image(np.ones((256, 256)) * 0.1) + input_image = Image(xp.ones((256, 256)) * 0.1) output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, np.ndarray) + + #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) + # Extending the test and setting the backend to torch @unittest.skipUnless(TORCH_AVAILABLE, "PyTorch is not installed.") class TestNoises_Torch(TestNoises_Numpy): BACKEND = "torch" pass - - def test_Gaussian(self): - noise = noises.Gaussian(mu=0.1, sigma=0.05) - input_image = torch.zeros((256, 256)) - output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, torch.Tensor) - self.assertEqual(output_image.shape, (256, 256)) - - def test_ComplexGaussian(self): - noise = noises.ComplexGaussian(mu=0.1, sigma=0.05) - input_image = torch.zeros((256, 256)) - output_image = noise.resolve(input_image) - self.assertIsInstance(output_image, torch.Tensor) - self.assertEqual(output_image.shape, (256, 256)) - self.assertTrue(torch.is_complex(output_image)) - - def test_Poisson(self): - noise = noises.Poisson(snr=20) - input_image = torch.ones((256, 256)) * 0.1 - output_image = noise.resolve(input_image) - - self.assertIsInstance(output_image, torch.Tensor) - self.assertEqual(output_image.shape, (256, 256)) - if __name__ == "__main__": unittest.main() From 55a74476b89b1c471f2224257f473a8e98e11110 Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 13:06:45 +0200 Subject: [PATCH 09/17] removed Image wrapper for Poisson --- deeptrack/tests/test_noises.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index e6049276f..6460a3bfe 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -66,7 +66,7 @@ def test_ComplexGaussian(self): def test_Poisson(self): noise = noises.Poisson(snr=20) - input_image = Image(xp.ones((256, 256)) * 0.1) + input_image = xp.ones((256, 256)) * 0.1 output_image = noise.resolve(input_image) #self.assertIsInstance(output_image, np.ndarray) From b4a09fbc9b684f2bee62aa753c5faebb4f7f6a4d Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 13:10:57 +0200 Subject: [PATCH 10/17] u --- deeptrack/tests/test_noises.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index 6460a3bfe..ed405694b 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -62,7 +62,7 @@ def test_ComplexGaussian(self): #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) - self.assertTrue(xp.iscomplexobj(output_image)) + self.assertTrue(xp.iscomplex(output_image)) def test_Poisson(self): noise = noises.Poisson(snr=20) From 9589b4dfaac9a978aeb9bbbd951cab554ade20f3 Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 13:14:35 +0200 Subject: [PATCH 11/17] complex --- deeptrack/tests/test_noises.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index ed405694b..64f49244d 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -62,7 +62,7 @@ def test_ComplexGaussian(self): #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) - self.assertTrue(xp.iscomplex(output_image)) + self.assertTrue(xp.any(xp.iscomplex(output_image))) def test_Poisson(self): noise = noises.Poisson(snr=20) From 8951fa7036138638603f1db35a35b2e551fb956b Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 13:16:41 +0200 Subject: [PATCH 12/17] u --- deeptrack/tests/test_noises.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index 64f49244d..e59805044 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -62,7 +62,7 @@ def test_ComplexGaussian(self): #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) - self.assertTrue(xp.any(xp.iscomplex(output_image))) + self.assertTrue(xp.any(xp.is_complex(output_image))) def test_Poisson(self): noise = noises.Poisson(snr=20) From 9a855b331c74396ebe9b1e21deb405ee5acefe6a Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 13:25:40 +0200 Subject: [PATCH 13/17] Update test_noises.py --- deeptrack/tests/test_noises.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index e59805044..c6a3a4f1e 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -62,7 +62,8 @@ def test_ComplexGaussian(self): #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) - self.assertTrue(xp.any(xp.is_complex(output_image))) + #TODO: Add xp function to check complex values + #self.assertTrue(xp.any(xp.is_complex(output_image))) def test_Poisson(self): noise = noises.Poisson(snr=20) From 19a93d1a39ef892dfdf32a511abfd96e12beee3f Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 13:33:48 +0200 Subject: [PATCH 14/17] Update test_noises.py --- deeptrack/tests/test_noises.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index c6a3a4f1e..6b50117e9 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -63,7 +63,7 @@ def test_ComplexGaussian(self): #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) #TODO: Add xp function to check complex values - #self.assertTrue(xp.any(xp.is_complex(output_image))) + self.assertTrue(xp.any(output_image.imag != 0)) def test_Poisson(self): noise = noises.Poisson(snr=20) From efd259533d380cce1afc915a697273e42008f37e Mon Sep 17 00:00:00 2001 From: Alex <95913221+Pwhsky@users.noreply.github.com> Date: Sun, 7 Sep 2025 13:38:08 +0200 Subject: [PATCH 15/17] whitespace --- deeptrack/tests/test_noises.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index 6b50117e9..bcc592f2e 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -59,7 +59,7 @@ def test_ComplexGaussian(self): noise = noises.ComplexGaussian(mu=0.1, sigma=0.05) input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - + #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) #TODO: Add xp function to check complex values @@ -69,7 +69,7 @@ def test_Poisson(self): noise = noises.Poisson(snr=20) input_image = xp.ones((256, 256)) * 0.1 output_image = noise.resolve(input_image) - + #self.assertIsInstance(output_image, np.ndarray) self.assertEqual(output_image.shape, (256, 256)) From 79ad87d3139abe86e353e200bbe51ed099a61340 Mon Sep 17 00:00:00 2001 From: mirjagranfors Date: Mon, 8 Sep 2025 09:11:48 +0200 Subject: [PATCH 16/17] added checks for array types --- deeptrack/tests/test_noises.py | 30 +++++++++++++++++++++--------- 1 file changed, 21 insertions(+), 9 deletions(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index bcc592f2e..db9c440fb 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -4,13 +4,12 @@ import unittest -import array_api_compat as apc import numpy as np from deeptrack.image import Image from deeptrack import noises -from deeptrack.backend import OPENCV_AVAILABLE, TORCH_AVAILABLE, xp +from deeptrack.backend import TORCH_AVAILABLE, xp from deeptrack.tests import BackendTestBase if TORCH_AVAILABLE: @@ -19,12 +18,21 @@ class TestNoises_Numpy(BackendTestBase): BACKEND = "numpy" + @property + def array_type(self): + if self.BACKEND == "numpy": + return np.ndarray + elif self.BACKEND == "torch": + return torch.Tensor + else: + raise ValueError(f"Unsupported backend: {self.BACKEND}") + def test_Offset(self): noise = noises.Offset(offset=0.5) input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - #self.assertIsInstance(output_image, np.ndarray) + self.assertIsInstance(output_image, self.array_type) self.assertEqual(output_image.shape, (256, 256)) self.assertTrue(xp.all(xp.asarray(output_image) == 0.5)) @@ -34,7 +42,7 @@ def test_Background(self): input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - #self.assertIsInstance(output_image, input_image) + self.assertIsInstance(output_image, self.array_type) self.assertEqual(output_image.shape, (256, 256)) self.assertTrue(xp.all(xp.asarray(output_image) == 0.5)) @@ -43,7 +51,7 @@ def test_Background(self): input_image = xp.ones((10, 10)) output_image = noise.resolve(input_image) - #self.assertIsInstance(output_image, input_image) + self.assertIsInstance(output_image, self.array_type) self.assertEqual(output_image.shape, (10, 10)) self.assertTrue(xp.all(xp.asarray(output_image) == 1.5)) @@ -52,7 +60,7 @@ def test_Gaussian(self): input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - #self.assertIsInstance(output_image, np.ndarray) + self.assertIsInstance(output_image, self.array_type) self.assertEqual(output_image.shape, (256, 256)) def test_ComplexGaussian(self): @@ -60,17 +68,21 @@ def test_ComplexGaussian(self): input_image = Image(xp.zeros((256, 256))) output_image = noise.resolve(input_image) - #self.assertIsInstance(output_image, np.ndarray) + self.assertIsInstance(output_image, self.array_type) self.assertEqual(output_image.shape, (256, 256)) - #TODO: Add xp function to check complex values self.assertTrue(xp.any(output_image.imag != 0)) + if self.BACKEND == "numpy": + self.assertTrue(np.iscomplexobj(output_image)) + elif self.BACKEND == "torch": + self.assertTrue(torch.is_complex(output_image)) + def test_Poisson(self): noise = noises.Poisson(snr=20) input_image = xp.ones((256, 256)) * 0.1 output_image = noise.resolve(input_image) - #self.assertIsInstance(output_image, np.ndarray) + self.assertIsInstance(output_image, self.array_type) self.assertEqual(output_image.shape, (256, 256)) From c4b98ccc6ad8f90ff5839ac3bc10cc79cf4a9e63 Mon Sep 17 00:00:00 2001 From: mirjagranfors Date: Mon, 8 Sep 2025 10:30:47 +0200 Subject: [PATCH 17/17] update test noises --- deeptrack/tests/test_noises.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deeptrack/tests/test_noises.py b/deeptrack/tests/test_noises.py index db9c440fb..0e15e800a 100644 --- a/deeptrack/tests/test_noises.py +++ b/deeptrack/tests/test_noises.py @@ -15,7 +15,7 @@ if TORCH_AVAILABLE: import torch -class TestNoises_Numpy(BackendTestBase): +class TestNoises_NumPy(BackendTestBase): BACKEND = "numpy" @property @@ -88,7 +88,7 @@ def test_Poisson(self): # Extending the test and setting the backend to torch @unittest.skipUnless(TORCH_AVAILABLE, "PyTorch is not installed.") -class TestNoises_Torch(TestNoises_Numpy): +class TestNoises_PyTorch(TestNoises_NumPy): BACKEND = "torch" pass