test for grubbs test implemented

test/funcs/test_data_modelling.py

@@ -4,18 +4,15 @@
 import pytest
 
 import numpy as np
-import pandas as pd
 
 from dios import dios
 from test.common import TESTFLAGGER
-import matplotlib.pyplot as plt
 
 from saqc.funcs.data_modelling import (
     modelling_polyFit
 )
-import numpy.polynomial.polynomial as poly
 
-TF=TESTFLAGGER[:1]
+TF = TESTFLAGGER[:1]
 @pytest.mark.parametrize("flagger", TF)
 @pytest.mark.parametrize("dat", [pytest.lazy_fixture("course_2")])
 def test_modelling_polyFit_forRegular(dat, flagger):

test/funcs/test_spikes_detection.py

@@ -12,7 +12,8 @@ from saqc.funcs.spikes_detection import (
     spikes_flagSlidingZscore,
     spikes_flagBasic,
     spikes_flagRaise,
-    spikes_flagMultivarScores
+    spikes_flagMultivarScores,
+    spikes_flagGrubbs
 )
 
 from test.common import TESTFLAGGER
@@ -120,3 +121,11 @@ def test_flagMultivarScores(dat, flagger):
         assert isflagged[characteristics['raise']].all()
         assert not isflagged[characteristics['return']].any()
         assert not isflagged[characteristics['drop']].any()
+
+@pytest.mark.parametrize("flagger", TESTFLAGGER)
+@pytest.mark.parametrize("dat", [pytest.lazy_fixture("course_3")])
+def test_grubbs(dat, flagger):
+    data, char_dict = dat(freq='10min', periods=45, initial_level=0, final_level=0, crowd_size=1, crowd_spacing=3, out_val=-10)
+    flagger = flagger.initFlags(data)
+    data, result_flagger = spikes_flagGrubbs(data, 'data', flagger, winsz=20, min_periods=15)
+    assert result_flagger.isFlagged('data')[char_dict["drop"]].all()
\ No newline at end of file