def test_simulate_rr(self):
"""Test oxi_peaks function"""
rr = simulate_rr(artefacts=True)
assert isinstance(rr, np.ndarray)
assert len(rr) == 350
def test_rr_artefacts(self):
rr = simulate_rr() # Import PPG recording
artefacts = rr_artefacts(rr)
artefacts = rr_artefacts(list(rr))
assert all(350 == x
for x in [len(artefacts[k]) for k in artefacts.keys()])
# Author: Nicolas Legrand <*****@*****.**>
import pandas as pd
import numpy as np
import unittest
from unittest import TestCase
from systole.plotly import plot_raw, plot_shortLong, plot_ectopic, \
plot_subspaces, plot_frequency, plot_nonlinear, plot_timedomain
from systole import import_ppg
from systole.utils import simulate_rr
rr = simulate_rr()
ppg = import_ppg()
signal_df = pd.DataFrame({
'time': np.arange(0, len(ppg[0])) / 75,
'signal': ppg[0]
})
class TestInteractive(TestCase):
def test_plot_raw(self):
"""Test plot_raw function"""
plot_raw(signal_df)
def test_plot_shortLong(self):
"""Test plot_shortLong function"""
plot_shortLong(rr)
def test_plot_ectopic(self):
"""Test plot_ectopic function"""
plot_ectopic(rr)
def test_rr_artefacts(self):
rr = simulate_rr() # Simulate RR time series
artefacts = rr_artefacts(rr)
artefacts = rr_artefacts(list(rr))
assert all(350 == x
for x in [len(artefacts[k]) for k in artefacts.keys()])