mean estimator pixel_size pass-through

j-c-c · j-c-c · commit 73a70a58db2b · 2025-06-16T13:44:21.000-04:00
diff --git a/src/aspire/reconstruction/estimator.py b/src/aspire/reconstruction/estimator.py
@@ -144,6 +144,7 @@ def estimate(self, b_coef=None, x0=None, tol=1e-5, regularizer=0):
             b_coef = self.src_backward()
         est_coef = self.conj_grad(b_coef, x0=x0, tol=tol, regularizer=regularizer)
         est = Coef(self.basis, est_coef).evaluate()
+        est.pixel_size = self.src.pixel_size
 
         return est
 
diff --git a/tests/test_mean_estimator.py b/tests/test_mean_estimator.py
@@ -56,6 +56,7 @@ def sim(L, dtype):
         ],
         dtype=dtype,
         seed=SEED,
+        pixel_size=1,
     )
 
     sim = sim.cache()  # precompute images
@@ -104,6 +105,9 @@ def test_estimate(sim, estimator, mask):
         est / np.linalg.norm(est), vol / np.linalg.norm(vol), atol=0.1
     )
 
+    # Check pixel_size pass-through
+    np.testing.assert_array_equal(sim.pixel_size, estimate.pixel_size)
+
 
 def test_adjoint(sim, basis, estimator):
     """

Original file line number	Diff line number	Diff line change
`@@ -56,6 +56,7 @@ def sim(L, dtype):`
`56`	`56`	`],`
`57`	`57`	`dtype=dtype,`
`58`	`58`	`seed=SEED,`
	`59`	`+ pixel_size=1,`
`59`	`60`	`)`
`60`	`61`
`61`	`62`	`sim = sim.cache() # precompute images`
`@@ -104,6 +105,9 @@ def test_estimate(sim, estimator, mask):`
`104`	`105`	`est / np.linalg.norm(est), vol / np.linalg.norm(vol), atol=0.1`
`105`	`106`	`)`
`106`	`107`
	`108`	`+ # Check pixel_size pass-through`
	`109`	`+ np.testing.assert_array_equal(sim.pixel_size, estimate.pixel_size)`
	`110`	`+`
`107`	`111`
`108`	`112`	`def test_adjoint(sim, basis, estimator):`
`109`	`113`	`"""`