Skip to content

mg/Update test_scatterers #426

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
giovannivolpe merged 2 commits into from
Sep 8, 2025
Merged

mg/Update test_scatterers #426

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
712af83
Update test_scatterers
mirjagranfors Sep 8, 2025
57d817e
update scatterers test
mirjagranfors Sep 8, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 33 additions & 11 deletions deeptrack/tests/test_scatterers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,17 +1,33 @@
import sys

sys.path.append(".") # Adds the module to path
# Use this only when running the test locally.
# import sys
# sys.path.append(".") # Adds the module to path

import unittest

from deeptrack import scatterers

import numpy as np

from deeptrack.optics import Fluorescence, Brightfield
from deeptrack.image import Image
from deeptrack import scatterers

from deeptrack.backend import TORCH_AVAILABLE, xp
from deeptrack.tests import BackendTestBase

if TORCH_AVAILABLE:
import torch


class TestScatterers_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}")

class TestScatterers(unittest.TestCase):
def test_PointParticle(self):
optics = Fluorescence(
NA=0.7,
Expand All @@ -27,7 +43,7 @@ def test_PointParticle(self):
)
imaged_scatterer = optics(scatterer)
output_image = imaged_scatterer.resolve()
self.assertIsInstance(output_image, np.ndarray)
self.assertIsInstance(output_image, self.array_type)
self.assertEqual(output_image.shape, (64, 64, 1))

def test_Ellipse(self):
Expand All @@ -48,7 +64,7 @@ def test_Ellipse(self):
)
imaged_scatterer = optics(scatterer)
output_image = imaged_scatterer.resolve()
self.assertIsInstance(output_image, np.ndarray)
self.assertIsInstance(output_image, self.array_type)
self.assertEqual(output_image.shape, (64, 64, 1))

def test_EllipseUpscale(self):
Expand Down Expand Up @@ -146,7 +162,7 @@ def test_Sphere(self):
)
imaged_scatterer = optics(scatterer)
output_image = imaged_scatterer.resolve()
self.assertIsInstance(output_image, np.ndarray)
self.assertIsInstance(output_image, self.array_type)
self.assertEqual(output_image.shape, (64, 64, 1))

def test_SphereUpscale(self):
Expand Down Expand Up @@ -188,7 +204,7 @@ def test_Ellipsoid(self):
)
imaged_scatterer = optics(scatterer)
output_image = imaged_scatterer.resolve()
self.assertIsInstance(output_image, np.ndarray)
self.assertIsInstance(output_image, self.array_type)
self.assertEqual(output_image.shape, (64, 64, 1))

def test_EllipsoidUpscale(self):
Expand Down Expand Up @@ -343,6 +359,12 @@ def test_MieStratifiedSphere(self):
imaged_scatterer_1 = optics_1(scatterer)
imaged_scatterer_1.update().resolve()

# TODO: Extending the test and setting the backend to torch
# @unittest.skipUnless(TORCH_AVAILABLE, "PyTorch is not installed.")
# class TestScatterers_PyTorch(TestScatterers_NumPy):
# BACKEND = "torch"
# pass


if __name__ == "__main__":
unittest.main()
Loading