diff --git a/test/funcs/test_soil_moisture_tests.py b/test/funcs/test_soil_moisture_tests.py index 0c8257fe368f895d9ee4856fe0df724fbbcee67f..e9834f2bf990d262fc3109015aceca48d12befa7 100644 --- a/test/funcs/test_soil_moisture_tests.py +++ b/test/funcs/test_soil_moisture_tests.py @@ -29,10 +29,10 @@ def test_flagSoilMoistureBySoilFrost(flagger): data, flagger_result = flagSoilMoistureBySoilFrost( data, "soil_moisture", flagger, "soil_temperature" ) - flag_assertion = np.arange(18, 37) + flag_assertion = np.arange(19, 37) flag_result = flagger_result.getFlags("soil_moisture") # .iloc[:, 0] - test_sum = (flag_result[flag_assertion] == flagger.BAD).sum() - assert test_sum == len(flag_assertion) + assert (flag_result[flag_assertion] == flagger.BAD).all() + @pytest.mark.parametrize("flagger", TESTFLAGGER) @@ -58,3 +58,6 @@ def test_flagSoilMoisturePrecipitationEvents(flagger): test_sum = (flag_result[flag_assertion] == flagger.BAD).sum() assert test_sum == len(flag_assertion) +if __name__ == "__main__": + flagger = TESTFLAGGER[2] + test_flagSoilMoistureBySoilFrost(flagger)