def test_tf_mxne():
"""Test convergence of TF-MxNE solver"""
alpha_space = 10
alpha_time = 5
M, G, active_set = _generate_tf_data()
X_hat, active_set_hat, E = tf_mixed_norm_solver(M, G,
alpha_space, alpha_time, maxit=200,
tol=1e-8, verbose=True,
n_orient=1, tstep=4, wsize=32)
assert_array_equal(np.where(active_set_hat)[0], active_set)
def test_tf_mxne_vs_mxne():
"""Test equivalence of TF-MxNE (with alpha_time=0) and MxNE"""
alpha_space = 60
alpha_time = 0
M, G, active_set = _generate_tf_data()
X_hat, active_set_hat, E = tf_mixed_norm_solver(M, G,
alpha_space, alpha_time, maxit=200,
tol=1e-8, verbose=True, debias=False,
n_orient=1, tstep=4, wsize=32)
# Also run L21 and check that we get the same
X_hat_l21, _, _ = mixed_norm_solver(M, G, alpha_space, maxit=200,
tol=1e-8, verbose=False, n_orient=1,
active_set_size=None, debias=False)
assert_array_almost_equal(X_hat, X_hat_l21, decimal=2)
