def test_signal_binarize():
signal = np.cos(np.linspace(start=0, stop=20, num=1000))
binary = nk.signal_binarize(signal)
assert len(binary) == 1000
binary = nk.signal_binarize(list(signal))
assert len(binary) == 1000
def test_signal_plot():
# Test with array
signal = nk.signal_simulate(duration=10, sampling_rate=1000)
nk.signal_plot(signal, sampling_rate=1000)
fig = plt.gcf()
for ax in fig.get_axes():
handles, labels = ax.get_legend_handles_labels()
assert labels == ['Signal']
assert len(labels) == len(handles) == len([signal])
assert ax.get_xlabel() == 'Time (s)'
plt.close(fig)
# Test with dataframe
data = pd.DataFrame({
"Signal2":
np.cos(np.linspace(start=0, stop=20, num=1000)),
"Signal3":
np.sin(np.linspace(start=0, stop=20, num=1000)),
"Signal4":
nk.signal_binarize(np.cos(np.linspace(start=0, stop=40, num=1000)))
})
nk.signal_plot(data, sampling_rate=None)
fig = plt.gcf()
for ax in fig.get_axes():
handles, labels = ax.get_legend_handles_labels()
assert labels == list(data.columns.values)
assert len(labels) == len(handles) == len(data.columns)
assert ax.get_xlabel() == 'Samples'
plt.close(fig)
# Test with list
signal = nk.signal_binarize(nk.signal_simulate(duration=10))
phase = nk.signal_phase(signal, method="percents")
nk.signal_plot([signal, phase])
fig = plt.gcf()
for ax in fig.get_axes():
handles, labels = ax.get_legend_handles_labels()
assert labels == ['Signal1', 'Signal2']
assert len(labels) == len(handles) == len([signal, phase])
assert ax.get_xlabel() == 'Samples'
plt.close(fig)