shithub: aubio

ref: 6bb268b1f41c1a8275b6da56f88630f4aa4cb5b6
dir: /python/tests/test_zero_crossing_rate.py/

#! /usr/bin/env python

from numpy.testing import TestCase

from aubio import fvec, zero_crossing_rate

buf_size = 2048

class zero_crossing_rate_test_case(TestCase):

    def setUp(self):
        self.vector = fvec(buf_size)

    def test_zeroes(self):
        """ check zero crossing rate on a buffer of 0. """
        self.assertEqual(0., zero_crossing_rate(self.vector))

    def test_ones(self):
        """ check zero crossing rate on a buffer of 1. """
        self.vector[:] = 1.
        self.assertEqual(0., zero_crossing_rate(self.vector))

    def test_impulse(self):
        """ check zero crossing rate on a buffer with an impulse """
        self.vector[buf_size / 2] = 1.
        self.assertEqual(0., zero_crossing_rate(self.vector))

    def test_negative_impulse(self):
        """ check zero crossing rate on a buffer with a negative impulse """
        self.vector[buf_size / 2] = -1.
        self.assertEqual(2./buf_size, zero_crossing_rate(self.vector))

    def test_single(self):
        """ check zero crossing rate on single crossing """
        self.vector[buf_size / 2 - 1] = 1.
        self.vector[buf_size / 2] = -1.
        self.assertEqual(2./buf_size, zero_crossing_rate(self.vector))

    def test_single_with_gap(self):
        """ check zero crossing rate on single crossing with a gap"""
        self.vector[buf_size / 2 - 2] = 1.
        self.vector[buf_size / 2] = -1.
        self.assertEqual(2./buf_size, zero_crossing_rate(self.vector))

if __name__ == '__main__':
    from unittest import main
    main()