package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-17 20:51:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

dsp/window: add Values.TransformTo window functions and example

Browse Source
This commit is contained in:
Dan Kortschak
2021-01-08 18:18:09 +10:30
parent 1c2011e56d
commit 676e41577b
3 changed files with 100 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
dsp/window/window_test.go
View File

@@ -174,7 +174,18 @@ func TestWindows(t *testing.T) {
src[i] = 1
}
dst = NewValues(test.fn, len(src)).Transform(src)
vals := NewValues(test.fn, len(src))
dst = vals.Transform(src)
if !floats.EqualApprox(dst, test.want, tol) {
t.Errorf("unexpected result for lookup window function %q:\ngot:%#.6v\nwant:%#.6v", test.name, dst, test.want)
}
for i := range src {
src[i] = 1
}
dst = make([]float64, len(src))
vals.TransformTo(dst, src)
if !floats.EqualApprox(dst, test.want, tol) {
t.Errorf("unexpected result for lookup window function %q:\ngot:%#.6v\nwant:%#.6v", test.name, dst, test.want)
}
@@ -202,7 +213,18 @@ func TestWindowsComplex(t *testing.T) {
src[i] = complex(1, 1)
}
dst = NewValues(test.fn, len(src)).TransformComplex(src)
vals := NewValues(test.fn, len(src))
dst = vals.TransformComplex(src)
if !equalApprox(dst, test.want, tol) {
t.Errorf("unexpected result for lookup window function %q:\ngot:%#.6v\nwant:%#.6v", test.name, dst, test.want)
}
for i := range src {
src[i] = complex(1, 1)
}
dst = make([]complex128, len(src))
vals.TransformComplexTo(dst, src)
if !equalApprox(dst, test.want, tol) {
t.Errorf("unexpected result for lookup window function %q:\ngot:%#.6v\nwant:%#.6v", test.name, dst, test.want)
}