import numpy as np
import pickle
from .testutil import datadir
from brainstat.mesh.data import mesh_normalize
[docs]def dummy_test(infile, expfile):
# load input test data
ifile = open(infile, "br")
idic = pickle.load(ifile)
ifile.close()
Y = idic["Y"]
mask = None
subdiv = "s"
if "mask" in idic.keys():
mask = idic["mask"]
# run mesh_normalize
Y_out, Yav_out = mesh_normalize(Y, mask, subdiv)
# load expected outout data
efile = open(expfile, "br")
expdic = pickle.load(efile)
efile.close()
exp_Y_out = expdic["Python_Y"]
exp_Yav_out = expdic["Python_Yav"]
testout = []
testout.append(np.allclose(Y_out, exp_Y_out, rtol=1e-05, equal_nan=True))
testout.append(np.allclose(Yav_out, exp_Yav_out, rtol=1e-05, equal_nan=True))
assert all(flag == True for (flag) in testout)
[docs]def test_01():
# testing only with mandatory input ['Y']
# ['Y'] : 2D np array, shape (1, 2), dtype('int64')
infile = datadir("statnor_01_IN.pkl")
expfile = datadir("statnor_01_OUT.pkl")
dummy_test(infile, expfile)
[docs]def test_02():
# testing with optional input ['mask']
# ['Y'] : 2D np array, shape (1, 10), dtype('int64')
# ['mask'] : 1D np array, shape (10,), dtype('bool')
infile = datadir("statnor_02_IN.pkl")
expfile = datadir("statnor_02_OUT.pkl")
dummy_test(infile, expfile)
[docs]def test_03():
# increase size of ['Y'], keep optional input ['mask']
# ['Y'] : 2D np array, shape (2, 10), dtype('int64')
# ['mask'] : 1D np array, shape (10,), dtype('bool')
infile = datadir("statnor_03_IN.pkl")
expfile = datadir("statnor_03_OUT.pkl")
dummy_test(infile, expfile)
[docs]def test_04():
# increase dimension of ['Y'], keep optional input ['mask']
# ['Y'] : 3D np array, shape (3, 4, 2), dtype('float64')
# ['mask'] : 1D np array, shape (4,), dtype('bool')
infile = datadir("statnor_04_IN.pkl")
expfile = datadir("statnor_04_OUT.pkl")
dummy_test(infile, expfile)