mirror of
https://github.com/macaodha/batdetect2.git
synced 2025-06-29 14:41:58 +02:00
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
"""Test suite for model functions."""
|
|
|
|
import warnings
|
|
from pathlib import Path
|
|
from typing import List
|
|
|
|
import numpy as np
|
|
from hypothesis import given, settings
|
|
from hypothesis import strategies as st
|
|
|
|
from batdetect2 import api
|
|
from batdetect2.detector import parameters
|
|
|
|
|
|
def test_can_import_model_without_warnings():
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("error")
|
|
api.load_model()
|
|
|
|
|
|
@settings(deadline=None, max_examples=5)
|
|
@given(duration=st.floats(min_value=0.1, max_value=2))
|
|
def test_can_import_model_without_pickle(duration: float):
|
|
# NOTE: remove this test once no other issues are found This is a temporary
|
|
# test to check that change in model loading did not impact model behaviour
|
|
# in any way.
|
|
|
|
samplerate = parameters.TARGET_SAMPLERATE_HZ
|
|
audio = np.random.rand(int(duration * samplerate))
|
|
|
|
model_without_pickle, model_params_without_pickle = api.load_model(
|
|
weights_only=True
|
|
)
|
|
model_with_pickle, model_params_with_pickle = api.load_model(
|
|
weights_only=False
|
|
)
|
|
|
|
assert model_params_without_pickle == model_params_with_pickle
|
|
|
|
predictions_without_pickle, _, _ = api.process_audio(
|
|
audio,
|
|
model=model_without_pickle,
|
|
)
|
|
predictions_with_pickle, _, _ = api.process_audio(
|
|
audio,
|
|
model=model_with_pickle,
|
|
)
|
|
|
|
assert predictions_without_pickle == predictions_with_pickle
|
|
|
|
|
|
def test_can_import_model_without_pickle_on_test_data(
|
|
example_audio_files: List[Path],
|
|
):
|
|
# NOTE: remove this test once no other issues are found This is a temporary
|
|
# test to check that change in model loading did not impact model behaviour
|
|
# in any way.
|
|
|
|
model_without_pickle, model_params_without_pickle = api.load_model(
|
|
weights_only=True
|
|
)
|
|
model_with_pickle, model_params_with_pickle = api.load_model(
|
|
weights_only=False
|
|
)
|
|
|
|
assert model_params_without_pickle == model_params_with_pickle
|
|
|
|
for audio_file in example_audio_files:
|
|
audio = api.load_audio(str(audio_file))
|
|
predictions_without_pickle, _, _ = api.process_audio(
|
|
audio,
|
|
model=model_without_pickle,
|
|
)
|
|
predictions_with_pickle, _, _ = api.process_audio(
|
|
audio,
|
|
model=model_with_pickle,
|
|
)
|
|
assert predictions_without_pickle == predictions_with_pickle
|