def test_three_pass(self):
sigma_thresh = 2
ser = pd.Series([-4, -4] + [0, 1, 2] * 4 + [5])
ser = sigma_edit_series(sigma_thresh, ser, max_iter=3)
self.assertTrue(all_in_bounds(sigma_thresh, ser))
def test_nothing_removed(self):
sigma_thresh = 10
ser = pd.Series([1, 2, 3, 4, 5])
ser = sigma_edit_series(sigma_thresh, ser, max_iter=1)
self.assertEqual(list(ser), [1, 2, 3, 4, 5])
def test_two_pass_exceed_iter(self):
sigma_thresh = 2
ser = pd.Series([-4] + [0, 1, 2] * 4 + [5])
with self.assertRaises(ValueError):
ser = sigma_edit_series(
sigma_thresh, ser, max_iter=1)
def test_empty_series(self):
sigma_thresh = 3
ser = pd.Series([])
with self.assertRaises(ValueError):
sigma_edit_series(sigma_thresh, ser)
def test_nan_series(self):
sigma_thresh = 3
ser = pd.Series([np.NaN for nn in range(10)])
with self.assertRaises(ValueError):
sigma_edit_series(sigma_thresh, ser)
def test_three_pass(self):
sigma_thresh = 2
ser = pd.Series([-4, -4] + [0, 1, 2] * 4 + [5])
ser = sigma_edit_series(sigma_thresh, ser, max_iter=3)
self.assertTrue(all_in_bounds(sigma_thresh, ser))
def test_two_pass_exceed_iter(self):
sigma_thresh = 2
ser = pd.Series([-4] + [0, 1, 2] * 4 + [5])
with self.assertRaises(ValueError):
ser = sigma_edit_series(sigma_thresh, ser, max_iter=1)
def test_nothing_removed(self):
sigma_thresh = 10
ser = pd.Series([1, 2, 3, 4, 5])
ser = sigma_edit_series(sigma_thresh, ser, max_iter=1)
self.assertEqual(list(ser), [1, 2, 3, 4, 5])
def test_nan_series(self):
sigma_thresh = 3
ser = pd.Series([np.NaN for nn in range(10)])
with self.assertRaises(ValueError):
sigma_edit_series(sigma_thresh, ser)
def test_empty_series(self):
sigma_thresh = 3
ser = pd.Series([])
with self.assertRaises(ValueError):
sigma_edit_series(sigma_thresh, ser)