import numpy as np
import pickle
from .testutil import datadir
from brainstat.mesh.utils import mesh_edges
[docs]def dummy_test(infile, expfile):
# load input test data
ifile = open(infile, "br")
idic = pickle.load(ifile)
ifile.close()
surf = {}
if "tri" in idic.keys():
surf["tri"] = idic["tri"]
if "lat" in idic.keys():
surf["lat"] = idic["lat"]
# run mesh_edges
out_edge = mesh_edges(surf)
# load expected outout data
efile = open(expfile, "br")
expdic = pickle.load(efile)
efile.close()
exp_edge = expdic["edg"]
testout = []
comp = np.allclose(out_edge, exp_edge, rtol=1e-05, equal_nan=True)
testout.append(comp)
assert all(flag == True for (flag) in testout)
# data *pkl consists of either keys ['tri'] or ['lat'], which will be assigned to
# the surf{} dictionary while testing
[docs]def test_01():
# ['tri'] is a 2D numpy array of shape (78, 3), dtype('float64')
infile = datadir("statedg_01_IN.pkl")
expfile = datadir("statedg_01_OUT.pkl")
dummy_test(infile, expfile)
[docs]def test_02():
# ['lat'] is a 2D numpy array of shape (10, 10), dtype('float64')
infile = datadir("statedg_02_IN.pkl")
expfile = datadir("statedg_02_OUT.pkl")
dummy_test(infile, expfile)
[docs]def test_03():
# ['lat'] is a 3D numpy array of shape (10, 10, 10), dtype('int64')
infile = datadir("statedg_03_IN.pkl")
expfile = datadir("statedg_03_OUT.pkl")
dummy_test(infile, expfile)
[docs]def test_04():
# ['tri'] is a 2D numpy array of shape (40960, 3), dtype('int32')
infile = datadir("statedg_04_IN.pkl")
expfile = datadir("statedg_04_OUT.pkl")
dummy_test(infile, expfile)
[docs]def test_06():
# ['tri'] is a 2D numpy array of shape (2044, 3), dtype('uint16')
infile = datadir("statedg_06_IN.pkl")
expfile = datadir("statedg_06_OUT.pkl")
dummy_test(infile, expfile)