From e259fb8c0e8c4ce46260244409c8db33900d21a8 Mon Sep 17 00:00:00 2001 From: Ananya-Joshi Date: Thu, 12 Nov 2020 00:09:35 -0500 Subject: [PATCH] Update validator/tests/test_checks.py Co-authored-by: nmdefries <42820733+nmdefries@users.noreply.github.com> --- validator/tests/test_checks.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/validator/tests/test_checks.py b/validator/tests/test_checks.py index f9a08b0f0..5ba0a0334 100644 --- a/validator/tests/test_checks.py +++ b/validator/tests/test_checks.py @@ -636,34 +636,34 @@ class TestDataCorrections: def test_pos_outlier(self): validator = Validator(self.params) - val = [30, 30.28571429, 30.57142857, 30.85714286, 31.14285714, + ref_val = [30, 30.28571429, 30.57142857, 30.85714286, 31.14285714, 31.42857143, 31.71428571, 32, 32, 32.14285714, 32.28571429, 32.42857143, 32.57142857, 32.71428571, 32.85714286, 33, 33, 33, 33, 33, 33, 33, 33, 33, 33, 33, 33.28571429, 33.57142857, 33.85714286, 34.14285714] - ref_val = [100, 100, 100] + test_val = [100, 100, 100] - data = {"val": val , "se": [np.nan] * len(val), - "sample_size": [np.nan] * len(val), "geo_id": ["1"] * len(val), - "time_value": pd.date_range(start="2020-09-24",end="2020-10-23")} ref_data = {"val": ref_val , "se": [np.nan] * len(ref_val), - "sample_size": [np.nan] * len(ref_val), "geo_id": ["1"] * len(ref_val), - "time_value": pd.date_range(start="2020-10-24",end="2020-10-26")} + "sample_size": [np.nan] * len(ref_val), "geo_id": ["1"] * len(ref_val), + "time_value": pd.date_range(start="2020-09-24", end="2020-10-23")} + test_data = {"val": test_val , "se": [np.nan] * len(test_val), + "sample_size": [np.nan] * len(test_val), "geo_id": ["1"] * len(test_val), + "time_value": pd.date_range(start="2020-10-24", end="2020-10-26")} - data2 = {"val": val , "se": [np.nan] * len(val), - "sample_size": [np.nan] * len(val), "geo_id": ["2"] * len(val), - "time_value": pd.date_range(start="2020-09-24",end="2020-10-23")} ref_data2 = {"val": ref_val , "se": [np.nan] * len(ref_val), "sample_size": [np.nan] * len(ref_val), "geo_id": ["2"] * len(ref_val), + "time_value": pd.date_range(start="2020-09-24", end="2020-10-23")} + test_data2 = {"val": test_val , "se": [np.nan] * len(test_val), + "sample_size": [np.nan] * len(test_val), "geo_id": ["2"] * len(test_val), "time_value": pd.date_range(start="2020-10-24",end="2020-10-26")} - test_df = pd.concat([pd.DataFrame(data), pd.DataFrame(data2)]) + test_df = pd.concat([pd.DataFrame(test_data), pd.DataFrame(test_data2)]) ref_df = pd.concat([pd.DataFrame(ref_data), pd.DataFrame(ref_data2)]) validator.check_positive_negative_spikes( - ref_df, test_df, "state", "signal") + test_df, ref_df, "state", "signal") assert len(validator.raised_errors) == 1