mirror of https://github.com/exaloop/codon.git
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
import numpy as np
|
|
from numpy import *
|
|
|
|
@test
|
|
def test_bartlett():
|
|
assert (round(np.bartlett(4), 8) == array([0., 0.66666667, 0.66666667,
|
|
0.])).all()
|
|
assert (np.bartlett(0) == np.empty(0)).all()
|
|
assert (np.bartlett(-7) == np.empty(0)).all()
|
|
assert (np.bartlett(1) == np.array([1.])).all()
|
|
|
|
# assert (round(np.bartlett(4.5), 8) == np.array([0., 0.57142857, 0.85714286, 0.28571429])).all()
|
|
|
|
test_bartlett()
|
|
|
|
@test
|
|
def test_blackman():
|
|
assert (np.blackman(-8) == np.empty(0)).all()
|
|
assert (np.blackman(1) == np.array([1.])).all()
|
|
x = np.blackman(3)
|
|
assert x[0] == -1.3877787807814457e-17
|
|
|
|
# a = np.blackman(6.6)
|
|
# assert a[6] == 0.01939475019748911
|
|
|
|
test_blackman()
|
|
|
|
@test
|
|
def test_hamming():
|
|
assert (np.hamming(-9) == np.empty(0)).all()
|
|
assert (np.hamming(1) == np.array([1.])).all()
|
|
x = np.hamming(12)
|
|
assert np.isclose(x[1], 0.15302337489765672)
|
|
assert np.isclose(x[6], 0.9813667678626689)
|
|
# a = np.hamming(6.6)
|
|
# assert np.isclose(a[0], 0.08)
|
|
|
|
test_hamming()
|
|
|
|
@test
|
|
def test_hanning():
|
|
assert (np.hanning(-9) == np.empty(0)).all()
|
|
assert (np.hanning(1) == np.array([1.])).all()
|
|
x = np.hanning(12)
|
|
assert np.isclose(x[1], 0.07937323358440945)
|
|
assert np.isclose(x[6], 0.9797464868072487)
|
|
# a = np.hanning(6.3)
|
|
# assert np.isclose(a[4], 0.4851833360887201)
|
|
|
|
test_hanning()
|
|
|
|
@test
|
|
def test_kaiser():
|
|
assert (np.kaiser(0, 0) == np.empty(0)).all()
|
|
assert (np.kaiser(0, 3) == np.empty(0)).all()
|
|
assert (np.kaiser(0, -4) == np.empty(0)).all()
|
|
assert (np.kaiser(1, -3) == np.array([1.])).all()
|
|
assert (np.kaiser(8, 0) == np.array([1.])).all()
|
|
assert (np.kaiser(1, 3) == np.array([1.])).all()
|
|
assert (np.kaiser(1, -3) == np.array([1.])).all()
|
|
assert (np.kaiser(-1, 0) == np.empty(0)).all()
|
|
assert (np.kaiser(-5, 7) == np.empty(0)).all()
|
|
assert (np.kaiser(-8, -4) == np.empty(0)).all()
|
|
x = np.kaiser(8, 4)
|
|
assert x[3] == 0.9652241821181469
|
|
y = np.kaiser(8, -5)
|
|
assert y[7] == 0.036710892271286676
|
|
assert (round(np.kaiser(4, 5), 8) == np.array(
|
|
[0.03671089, 0.77532210, 0.77532210, 0.03671089])).all()
|
|
assert (round(np.kaiser(4, 6), 8) == np.array(
|
|
[0.01487334, 0.73189542, 0.73189542, 0.01487334])).all()
|
|
|
|
# assert (round(np.kaiser(4, 8.6), 8) == np.array([0.00133251, 0.63041193, 0.63041193, 0.00133251])).all()
|
|
|
|
test_kaiser()
|