1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
/* Copyright 2016. Martin Uecker.
* All rights reserved. Use of this source code is governed by
* a BSD-style license which can be found in the LICENSE file.
*
* Authors:
* 2016 Martin Uecker <martin.uecker@med.uni-goettingen.de>
*/
#include <stdlib.h>
#include "lowrank/batchsvd.h"
#include "num/flpmath.h"
#include "misc/debug.h"
#include "misc/misc.h"
#include "utest.h"
static bool test_batch_svthresh_tall(void)
{
complex float inout[1][5][3] = { {
{ 3., 0., 0. },
{ 0., 2., 0. },
{ 0., 0., 1. },
{ 0., 0., 0. },
{ 0., 0., 0. },
} };
batch_svthresh(3, 5, 1, 1., inout);
const complex float ref[1][5][3] = { {
{ 2., 0., 0. },
{ 0., 1., 0. },
{ 0., 0., 0. },
{ 0., 0., 0. },
{ 0., 0., 0. },
} };
long dims[3] = { 3, 5, 1 };
UT_ASSERT(md_znrmse(3, dims, &ref[0][0][0], &inout[0][0][0]) < UT_TOL);
return true;
}
static bool test_batch_svthresh_wide(void)
{
complex float inout[1][3][5] = { {
{ 3., 0., 0., 0., 0. },
{ 0., 2., 0., 0., 0. },
{ 0., 0., 1., 0., 0. },
} };
batch_svthresh(5, 3, 1, 1., inout);
const complex float ref[1][3][5] = { {
{ 2., 0., 0., 0., 0. },
{ 0., 1., 0., 0., 0. },
{ 0., 0., 0., 0., 0. },
} };
long dims[3] = { 5, 3, 1 };
UT_ASSERT(md_znrmse(3, dims, &ref[0][0][0], &inout[0][0][0]) < UT_TOL);
return true;
}
UT_REGISTER_TEST(test_batch_svthresh_tall);
UT_REGISTER_TEST(test_batch_svthresh_wide);
|
