Merge pull request #77 from robmarkcole/update-mqtt

[EVA-2020-02-2.git] / library / tests / test_noise.py

import pytest\r
\r
\r
def test_noise_setup(sounddevice, numpy):\r
    from enviroplus.noise import Noise\r
\r
    noise = Noise(sample_rate=16000, duration=0.1)\r
    del noise\r
\r
\r
def test_noise_get_amplitudes_at_frequency_ranges(sounddevice, numpy):\r
    from enviroplus.noise import Noise\r
\r
    noise = Noise(sample_rate=16000, duration=0.1)\r
    noise.get_amplitudes_at_frequency_ranges([\r
        (100, 500),\r
        (501, 1000)\r
    ])\r
\r
    sounddevice.rec.assert_called_with(0.1 * 16000, samplerate=16000, blocking=True, channels=1, dtype='float64')\r
\r
\r
def test_noise_get_noise_profile(sounddevice, numpy):\r
    from enviroplus.noise import Noise\r
\r
    numpy.mean.return_value = 10.0\r
\r
    noise = Noise(sample_rate=16000, duration=0.1)\r
    amp_low, amp_mid, amp_high, amp_total = noise.get_noise_profile(\r
        noise_floor=100,\r
        low=0.12,\r
        mid=0.36,\r
        high=None)\r
\r
    sounddevice.rec.assert_called_with(0.1 * 16000, samplerate=16000, blocking=True, channels=1, dtype='float64')\r
\r
    assert amp_total == 10.0\r
\r
\r
def test_get_amplitude_at_frequency_range(sounddevice, numpy):\r
    from enviroplus.noise import Noise\r
\r
    noise = Noise(sample_rate=16000, duration=0.1)\r
\r
    noise.get_amplitude_at_frequency_range(0, 8000)\r
\r
    with pytest.raises(ValueError):\r
        noise.get_amplitude_at_frequency_range(0, 16000)\r