pytfd_compat/tests/test_validation.py

17 lines
602 B
Python

import numpy as np
from validation.compare import compare_stft_to_scipy, make_validation_signals
def test_validation_signals_present():
signals = make_validation_signals()
assert {"dual_tone_impulse", "linear_chirp", "click_train"} <= set(signals)
def test_scipy_alignment_validation_small_signal():
signal = make_validation_signals()["dual_tone_impulse"]
metrics = compare_stft_to_scipy(signal, window_name="hanning", window_length=33, hop=8, n_fft=128)
assert metrics["best_offset"] >= 0
assert metrics["max_abs_diff"] < 1e-10
assert metrics["mean_abs_diff"] < 1e-12