kopia lustrzana https://github.com/corrscope/corrscope
57 wiersze
1.3 KiB
Python
57 wiersze
1.3 KiB
Python
import matplotlib.pyplot as plt
|
|
import pytest
|
|
from matplotlib.axes import Axes
|
|
from matplotlib.figure import Figure
|
|
|
|
from ovgenpy import triggers
|
|
from ovgenpy.triggers import CorrelationTrigger
|
|
from ovgenpy.wave import Wave
|
|
|
|
|
|
triggers.SHOW_TRIGGER = False
|
|
|
|
|
|
@pytest.fixture(scope='session', params=[False, True])
|
|
def cfg(request):
|
|
use_edge_trigger = request.param
|
|
return CorrelationTrigger.Config(
|
|
trigger_strength=1,
|
|
use_edge_trigger=use_edge_trigger,
|
|
|
|
responsiveness=1,
|
|
falloff_width=2,
|
|
)
|
|
|
|
|
|
def test_trigger(cfg):
|
|
# wave = Wave(None, 'tests/sine440.wav')
|
|
wave = Wave(None, 'tests/impulse24000.wav')
|
|
|
|
iters = 5
|
|
plot = False
|
|
x = 24000 - 500
|
|
trigger = cfg(wave, 4000)
|
|
|
|
if plot:
|
|
BIG = 0.95
|
|
SMALL = 0.05
|
|
fig, axes = plt.subplots(iters, gridspec_kw=dict(
|
|
top=BIG, right=BIG,
|
|
bottom=SMALL, left=SMALL,
|
|
)) # type: Figure, Axes
|
|
fig.tight_layout()
|
|
else:
|
|
axes = range(iters)
|
|
|
|
for i, ax in enumerate(axes):
|
|
if i:
|
|
offset2 = trigger.get_trigger(x)
|
|
print(offset2)
|
|
assert offset2 == 24000
|
|
if plot:
|
|
ax.plot(trigger._buffer, label=str(i))
|
|
ax.grid()
|
|
|
|
if plot:
|
|
plt.show()
|