#! /usr/bin/env python

from numpy.testing import TestCase
from numpy.testing.utils import assert_equal, assert_almost_equal
from numpy import cos, arange
from math import pi

from aubio import window, level_lin, db_spl, silence_detection, level_detection

from aubio import fvec, float_type

class aubio_window(TestCase):

    def test_accept_name_and_size(self):
        window("default", 1024)

    def test_fail_name_not_string(self):
        with self.assertRaises(TypeError):
            window(10, 1024)

    def test_fail_size_not_int(self):
        with self.assertRaises(TypeError):
            window("default", "default")

    def test_compute_hanning_1024(self):
        size = 1024
        aubio_window = window("hanning", size)
        numpy_window = .5 - .5 * cos(2. * pi * arange(size) / size)
        assert_almost_equal(aubio_window, numpy_window)

class aubio_level_lin(TestCase):
    def test_accept_fvec(self):
        level_lin(fvec(1024))

    def test_fail_not_fvec(self):
        try:
            level_lin("default")
        except ValueError as e:
            pass
        else:
            self.fail('non-number input phase does not raise a TypeError')

    def test_zeros_is_zeros(self):
        assert_equal(level_lin(fvec(1024)), 0.)

    def test_minus_ones_is_one(self):
        from numpy import ones
        assert_equal(level_lin(-ones(1024, dtype = float_type)), 1.)

class aubio_db_spl(TestCase):
    def test_accept_fvec(self):
        db_spl(fvec(1024))

    def test_fail_not_fvec(self):
        try:
            db_spl("default")
        except ValueError as e:
            pass
        else:
            self.fail('non-number input phase does not raise a TypeError')

    def test_zeros_is_inf(self):
        from math import isinf
        assert isinf(db_spl(fvec(1024)))

    def test_minus_ones_is_zero(self):
        from numpy import ones
        assert_equal(db_spl(-ones(1024, dtype = float_type)), 0.)

class aubio_silence_detection(TestCase):
    def test_accept_fvec(self):
        silence_detection(fvec(1024), -70.)

    def test_fail_not_fvec(self):
        try:
            silence_detection("default", -70)
        except ValueError as e:
            pass
        else:
            self.fail('non-number input phase does not raise a TypeError')

    def test_zeros_is_one(self):
        from math import isinf
        assert silence_detection(fvec(1024), -70) == 1

    def test_minus_ones_is_zero(self):
        from numpy import ones
        assert silence_detection(ones(1024, dtype = float_type), -70) == 0

class aubio_level_detection(TestCase):
    def test_accept_fvec(self):
        level_detection(fvec(1024), -70.)

    def test_fail_not_fvec(self):
        try:
            level_detection("default", -70)
        except ValueError as e:
            pass
        else:
            self.fail('non-number input phase does not raise a TypeError')

    def test_zeros_is_one(self):
        from math import isinf
        assert level_detection(fvec(1024), -70) == 1

    def test_minus_ones_is_zero(self):
        from numpy import ones
        assert level_detection(ones(1024, dtype = float_type), -70) == 0

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