added tests for interval integration

douglasrizzo · Mar 5, 2016 · 9c9bae8 · 9c9bae8
1 parent e484df7
commit 9c9bae8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/catsim/selection.py b/catsim/selection.py
@@ -697,7 +697,7 @@ class IntervalIntegrationSelector(Selector):
     .. math:: argmax_{i \\in I} \\int_{\\hat\\theta - \\delta}^{\\hat\\theta - \\delta}I_i(\\hat\\theta)
 
     :param interval: the interval of the integral. If no interval is passed, the
-                     integral is calculated from :math:`[-\\inf, \\inf]`.
+                     integral is calculated from :math:`[-\\infty, \\infty]`.
     """
 
     def __init__(self, interval: float=None):

diff --git a/catsim/testing.py b/catsim/testing.py
@@ -7,7 +7,7 @@
 from catsim.simulation import Simulator
 from catsim.cat import generate_item_bank
 from catsim.stopping import MaxItemStopper, MinErrorStopper
-from catsim.selection import MaxInfoSelector, ClusterSelector, LinearSelector, RandomSelector, AStratifiedBBlockingSelector, MaxInfoStratificationSelector, MaxInfoBBlockingSelector, AStratifiedSelector, The54321Selector, RandomesqueSelector
+from catsim.selection import MaxInfoSelector, ClusterSelector, LinearSelector, RandomSelector, AStratifiedBBlockingSelector, MaxInfoStratificationSelector, MaxInfoBBlockingSelector, AStratifiedSelector, The54321Selector, RandomesqueSelector, IntervalIntegrationSelector
 from catsim.initialization import RandomInitializer, FixedPointInitializer
 from catsim.estimation import HillClimbingEstimator, DifferentialEvolutionEstimator, FMinEstimator
 
@@ -96,7 +96,7 @@ def test_simulations():
                           ),
         FixedPointInitializer(0)
     ]
-    infinite_selectors = [MaxInfoSelector(), RandomSelector()]
+    infinite_selectors = [MaxInfoSelector(), RandomSelector(), IntervalIntegrationSelector(0.3)]
     finite_selectors = [
         LinearSelector(
             numpy.random.randint(