Wrote min_ind_range function

Author: kaashmonee

unit_tests.py

@@ -23,6 +23,9 @@ def test_sigma_clipping_no_failures(self):
         self.assertTrue(new_xvals.all() == xvalues.all()
                         and new_yvals.all() == yvalues.all())
 
+
+        print("sigma_clipping no failures passed...")
+
 
     def test_sortxy(self):
         """
@@ -41,6 +44,8 @@ def test_sortxy(self):
         self.assertTrue(list(sortedx) == list(range(1, 101)))
         self.assertTrue(list(sortedy) == list(range(401, 501)))
 
+        print("sortxy() passed...")
+
 
 
     def test_sigma_clipping_failures1(self):
@@ -62,6 +67,8 @@ def test_sigma_clipping_failures1(self):
         self.assertTrue(np.array_equal(xvalues, new_xvals))
         self.assertTrue(np.array_equal(yvalues, new_yvals))
 
+        print("sigma_clipping_failures1 passed...")
+
     def test_sigma_clipping_failures2(self):
         """
         Tests sigma clipping function on failing the first and last elements.
@@ -81,33 +88,25 @@ def test_sigma_clipping_failures2(self):
         self.assertTrue(np.array_equal(xvalues, new_xvals))
         self.assertTrue(np.array_equal(yvalues, new_yvals))
 
-    # def test_sigma_clipping_failures3(self):
-    #     """
-    #     This ensures that the alert message is displayed if over 10% of 
-    #     the pixels are removed.
-    #     """
-    #     xvalues = np.arange(10)
-    #     yvalues = np.random.normal(10, 3, 10)
-    #     yvalues[3] = -5
-    #     yvalues[5] = 100
-
-    #     new_xvals, new_yvals = util.sigma_clip(xvalues, yvalues)
-
-    #     xvalues = np.delete(xvalues, [3, 5])
-    #     yvalues = np.delete(yvalues, [3, 5])
-
-    #     message = "Over 10% of pixels have been rejected in the sigma_clip routine."
-
-    #     expected = """
-    #     ==========================
-    #     Warning: %s
-    #     ==========================
-    #     """
-    #     captured = sys.__stdout__
-    #     print(sys.__stdout__)
-    #     print(expected % message)
-
-    #     self.assertEqual(expected, captured)
+        print("sigma_clipping_failures2 passed...")
+
+    
+    def test_min_ind_range(self):
+        test_array1 = [1, 3, 4, 5, 9, 0, 3, 4, -1, 30]
+        start, end = 2, 5
+        min_ind = util.min_ind_range(test_array1, start, end)
+        self.assertTrue(min_ind == 2)
+
+        start, end = 0, len(test_array1)//2
+        min_ind = util.min_ind_range(test_array1, start, end)
+        self.assertTrue(min_ind == 0)
+
+        start, end = len(test_array1)//2, len(test_array1)
+        min_ind = util.min_ind_range(test_array1, start, end)
+        self.assertTrue(min_ind == 8)
+
+        print("min_ind_range() tests passed...")
+
 
 
 if __name__ == "__main__":

util.py

@@ -199,6 +199,15 @@ def plot(self):
             plt.plot(x, y, color=color)
 
 
+def min_ind_range(array, start, end):
+    """
+    Obtains the indices of the absolute minima of the first parameter
+    in the range provided by the 2nd and 3rd parameter.
+    """
+    temp = array[start:end]
+    min_ind_temp = np.argmin(temp)
+    min_ind_real = start + min_ind_temp
+    return min_ind_real
 
 
 def get_vmin_vmax(image):