fix unittest for LSVRG

TomographicImaging · Jul 21, 2024 · c6d7f29 · c6d7f29
1 parent 7f16acb
commit c6d7f29
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/Wrappers/Python/test/test_approximate_gradient.py b/Wrappers/Python/test/test_approximate_gradient.py
@@ -531,4 +531,14 @@ def test_LSVRG_toy_example_store_gradients(self):
         self.assertNumpyArrayAlmostEqual(
             alg_stochastic.x.as_array(), u_cvxpy.value, 3)
         self.assertNumpyArrayAlmostEqual(
-            alg_stochastic.x.as_array(), b.as_array(), 3)
+            alg_stochastic.x.as_array(), b.as_array(), 3)
+
+    def test_sampler_out_of_range(self):
+        def g(index):
+            return -2
+        bad_sampler = Sampler.from_function(12,g)
+        f = self.stochastic_estimator([self.f]*10, bad_sampler, seed=3)
+        with self.assertRaises(IndexError):
+            f.gradient(self.initial)
+            f.gradient(self.initial)
+