Skip to content
Snippets Groups Projects
Commit 4317e5e6 authored by Peter Lünenschloß's avatar Peter Lünenschloß
Browse files

constants test test added

parent 4870a750
No related branches found
No related tags found
No related merge requests found
#! /usr/bin/env python
# -*- coding: utf-8 -*-
import pytest
import numpy as np
import pandas as pd
from saqc.flagger.baseflagger import BaseFlagger
from saqc.flagger.dmpflagger import DmpFlagger
from saqc.flagger.simpleflagger import SimpleFlagger
from saqc.funcs.constants_detection import flagConstants_VarianceBased
from saqc.lib.tools import getPandasData
TESTFLAGGERS = [
BaseFlagger(['NIL', 'GOOD', 'BAD']),
DmpFlagger(),
SimpleFlagger()]
@pytest.fixture(scope='module')
def constants_data():
index = pd.date_range(start='1.1.2011 00:00:00', end='2.1.2011 00:03:00', freq='5min')
constants_series = pd.Series(np.linspace(-50, 50, index.size), index=index, name='constants_data')
constants_series.iloc[5:25] = 0
# constants_series.iloc[1000] = -100
flag_assertion = list(range(5, 25))
return constants_series, flag_assertion
@pytest.mark.parametrize('flagger', TESTFLAGGERS)
def test_flagConstants_VarianceBased(constants_data, flagger):
data = constants_data[0]
flags = flagger.initFlags(data)
data, flag_result = flagConstants_VarianceBased(data, flags, 'constants_data', flagger, plateau_window_min='1h')
flag_result = getPandasData(flag_result, 0)
test_sum = (flag_result[constants_data[1]] == flagger.BAD).sum()
assert test_sum == len(constants_data[1])
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment