def test_outliers2(self):
"""Test that outlier detection works."""
series = np.copy(self.series)
series[10] = -2
with pytest.warns(UserWarning) as record:
series2 = purge_outliers(series)
assert "Found 1 outliers" in record[0].message.args[0]
assert np.all(series2[:10] == series[:10])
assert np.all(series2[11:] == series[11:])
np.testing.assert_almost_equal(series2[10],
(series[9] + series[11]) / 2)
def test_outliers3(self):
"""Test that outlier detection works."""
series = np.copy(self.series)
series[10] = 20
series[11] = 20
with pytest.warns(UserWarning) as record:
series2 = purge_outliers(series)
assert "Found 2 outliers" in record[0].message.args[0]
assert np.all(series2[:10] == series[:10])
assert np.all(series2[12:] == series[12:])
lower = np.min([series[9], series[12]])
upper = np.max([series[9], series[12]])
assert np.all((series2[10:12] > lower) & (series2[10:12] < upper))