test_heuristics.py

import pathlib

import bletl

dir_testfiles = pathlib.Path(pathlib.Path(__file__).absolute().parent, "data")
FP_TESTFILE = pathlib.Path(dir_testfiles, "BLPro", "107-AR_Coryne-AR-2019-04-15-12-39-30.csv")


class TestDOPeakDetection:
    def test_find_peak(self):
        bldata = bletl.parse(FP_TESTFILE)

        x, y = bldata["DO"].get_timeseries("A01")

        c_peak = bletl.find_do_peak(
            x, y, delay_a=0.5, threshold_a=70, delay_b=0, threshold_b=80, initial_delay=1
        )

        assert c_peak == 60
        return

    def test_find_no_peak_with_long_delay_a(self):
        bldata = bletl.parse(FP_TESTFILE)

        x, y = bldata["DO"].get_timeseries("A01")

        # The DO was below 70 for only ~2 h
        c_peak = bletl.find_do_peak(
            x, y, delay_a=5, threshold_a=70, delay_b=0, threshold_b=80, initial_delay=1
        )

        assert c_peak is None
        return

    def test_find_no_peak_with_long_delay_b(self):
        bldata = bletl.parse(FP_TESTFILE)

        x, y = bldata["DO"].get_timeseries("A01")

        # The series continues for only ~9.5 h after the peak
        c_peak = bletl.find_do_peak(
            x, y, delay_a=0.5, threshold_a=70, delay_b=12, threshold_b=80, initial_delay=1
        )

        assert c_peak is None
        return

    def test_find_no_peak_with_long_initial_delay(self):
        bldata = bletl.parse(FP_TESTFILE)

        x, y = bldata["DO"].get_timeseries("A01")

        # The peak is within the first ~12 h
        c_peak = bletl.find_do_peak(
            x, y, delay_a=0.5, threshold_a=70, delay_b=4, threshold_b=80, initial_delay=15
        )

        assert c_peak is None
        return