corrscope/tests/test_channel.py

import numpy as np
import pytest
from hypothesis import given, reproduce_failure
from hypothesis.strategies import integers
from pytest_mock import MockFixture

import ovgenpy.channel
import ovgenpy.ovgenpy
from ovgenpy.channel import ChannelConfig, Channel
from ovgenpy.ovgenpy import default_config, Ovgen, BenchmarkMode
from ovgenpy.triggers import NullTriggerConfig

assert reproduce_failure


positive = integers(min_value=1)

@given(subsampling=positive, trigger_width_ratio=positive, render_width_ratio=positive)
def test_channel_subsampling(
    subsampling: int,
    trigger_width_ratio: int,
    render_width_ratio: int,
    mocker: MockFixture
):
    """ Ensure nsamp and trigger/render subsampling are computed correctly. """

    ovgenpy.ovgenpy.RENDER_PROFILING = False    # Cleanup Hypothesis testing logs

    Wave = mocker.patch.object(ovgenpy.channel, 'Wave')
    wave = Wave.return_value

    def get_around(sample: int, region_nsamp: int, subsampling: int):
        return np.zeros(region_nsamp)

    wave.get_around.side_effect = get_around
    wave.nsamp = 10000
    wave.smp_s = 48000

    ccfg = ChannelConfig(
        'tests/sine440.wav',
        trigger_width_ratio=trigger_width_ratio,
        render_width_ratio=render_width_ratio,
    )
    cfg = default_config(
        channels=[ccfg],
        subsampling=subsampling,
        trigger=NullTriggerConfig(),
        outputs=[],
        benchmark_mode=BenchmarkMode.OUTPUT
    )
    channel = Channel(ccfg, cfg)

    # Ensure channel.nsamp, trigger_subsampling, render_subsampling are correct.
    ideal_nsamp = pytest.approx(
        round(cfg.render_width_s * channel.wave.smp_s / subsampling), 1)

    assert channel.nsamp == ideal_nsamp
    assert channel.trigger_subsampling == subsampling * trigger_width_ratio
    assert channel.render_subsampling == subsampling * render_width_ratio

    # Ensure trigger uses channel.nsamp and trigger_subsampling.
    trigger = channel.trigger
    assert trigger._nsamp == channel.nsamp
    assert trigger._subsampling == channel.trigger_subsampling

    # Ensure ovgenpy calls render using channel.nsamp and render_subsampling.
    ovgen = Ovgen(cfg)
    renderer = mocker.patch.object(Ovgen, '_load_renderer').return_value
    ovgen.play()

    # Inspect arguments to wave.get_around()
    (_sample, _region_nsamp, _subsampling), kwargs = wave.get_around.call_args
    assert _region_nsamp == channel.nsamp
    assert _subsampling == channel.render_subsampling

    # Inspect arguments to renderer.render_frame()
    # datas: List[np.ndarray]
    (datas,), kwargs = renderer.render_frame.call_args
    render_data = datas[0]
    assert len(render_data) == channel.nsamp


# line_color is tested in test_renderer.py
# todo test ChannelConfig.ampl_ratio