def test_saveload(self):
with tempfile.TemporaryDirectory() as tdir:
fname = os.path.join(tdir, "dummy")
# basic but most important: ensure object integrity is preserved
checkAttr = [
"data", "dimord", "sampleinfo", "samplerate", "trialinfo"
]
dummy = EventData(self.data, samplerate=10)
dummy.save(fname)
filename = construct_spy_filename(fname, dummy)
dummy2 = load(filename)
for attr in checkAttr:
assert np.array_equal(getattr(dummy, attr),
getattr(dummy2, attr))
dummy3 = load(fname)
for attr in checkAttr:
assert np.array_equal(getattr(dummy3, attr),
getattr(dummy, attr))
save(dummy3, container=os.path.join(tdir, "ymmud"))
dummy4 = load(os.path.join(tdir, "ymmud"))
for attr in checkAttr:
assert np.array_equal(getattr(dummy4, attr),
getattr(dummy, attr))
del dummy2, dummy3, dummy4 # avoid PermissionError in Windows
# overwrite existing file w/new data
dummy.samplerate = 20
dummy.save()
dummy2 = load(filename=filename)
assert dummy2.samplerate == 20
del dummy, dummy2
time.sleep(0.1) # wait to kick-off garbage collection
# ensure trialdefinition is saved and loaded correctly
dummy = EventData(self.data,
trialdefinition=self.trl,
samplerate=10)
dummy.save(fname, overwrite=True)
dummy2 = load(filename)
assert np.array_equal(dummy.sampleinfo, dummy2.sampleinfo)
assert np.array_equal(dummy._t0, dummy2._t0)
assert np.array_equal(dummy.trialinfo, dummy2.trialinfo)
del dummy, dummy2
# swap dimensions and ensure `dimord` is preserved
dummy = EventData(self.data,
dimord=["eventid", "sample"],
samplerate=10)
dummy.save(fname + "_dimswap")
filename = construct_spy_filename(fname + "_dimswap", dummy)
dummy2 = load(filename)
assert dummy2.dimord == dummy.dimord
assert dummy2.eventid.size == self.num_smp # swapped
assert dummy2.data.shape == dummy.data.shape
# Delete all open references to file objects b4 closing tmp dir
del dummy, dummy2
time.sleep(0.1)
def test_saveload(self):
with tempfile.TemporaryDirectory() as tdir:
fname = os.path.join(tdir, "dummy")
# basic but most important: ensure object integrity is preserved
checkAttr = [
"channel", "data", "dimord", "freq", "sampleinfo",
"samplerate", "taper", "trialinfo"
]
dummy = SpectralData(self.data, samplerate=1000)
dummy.save(fname)
filename = construct_spy_filename(fname, dummy)
# dummy2 = SpectralData(filename)
# for attr in checkAttr:
# assert np.array_equal(getattr(dummy, attr), getattr(dummy2, attr))
dummy3 = load(fname)
for attr in checkAttr:
assert np.array_equal(getattr(dummy3, attr),
getattr(dummy, attr))
save(dummy3, container=os.path.join(tdir, "ymmud"))
dummy4 = load(os.path.join(tdir, "ymmud"))
for attr in checkAttr:
assert np.array_equal(getattr(dummy4, attr),
getattr(dummy, attr))
del dummy, dummy3, dummy4 # avoid PermissionError in Windows
# ensure trialdefinition is saved and loaded correctly
dummy = SpectralData(self.data,
trialdefinition=self.trl,
samplerate=1000)
dummy.save(fname, overwrite=True)
dummy2 = load(filename)
assert np.array_equal(dummy.trialdefinition,
dummy2.trialdefinition)
# test getters
assert np.array_equal(dummy.sampleinfo, dummy2.sampleinfo)
assert np.array_equal(dummy._t0, dummy2._t0)
assert np.array_equal(dummy.trialinfo, dummy2.trialinfo)
# swap dimensions and ensure `dimord` is preserved
dummy = SpectralData(self.data,
dimord=["time", "channel", "taper", "freq"],
samplerate=1000)
dummy.save(fname + "_dimswap")
filename = construct_spy_filename(fname + "_dimswap", dummy)
dummy2 = load(filename)
assert dummy2.dimord == dummy.dimord
assert dummy2.channel.size == self.nt # swapped
assert dummy2.taper.size == self.nf # swapped
assert dummy2.data.shape == dummy.data.shape
# Delete all open references to file objects b4 closing tmp dir
del dummy, dummy2
def test_saveload(self):
with tempfile.TemporaryDirectory() as tdir:
fname = os.path.join(tdir, "dummy")
# basic but most important: ensure object integrity is preserved
checkAttr = [
"channel", "data", "dimord", "sampleinfo", "samplerate",
"trialinfo"
]
dummy = AnalogData(data=self.data, samplerate=1000)
dummy.save(fname)
filename = construct_spy_filename(fname, dummy)
# NOTE: We removed support for loading data via the constructor
# dummy2 = AnalogData(filename)
# for attr in checkAttr:
# assert np.array_equal(getattr(dummy, attr), getattr(dummy2, attr))
dummy3 = load(fname)
for attr in checkAttr:
assert np.array_equal(getattr(dummy3, attr),
getattr(dummy, attr))
save(dummy3, container=os.path.join(tdir, "ymmud"))
dummy4 = load(os.path.join(tdir, "ymmud"))
for attr in checkAttr:
assert np.array_equal(getattr(dummy4, attr),
getattr(dummy, attr))
del dummy, dummy3, dummy4 # avoid PermissionError in Windows
# # FIXME: either remove or repair this
# # save object hosting VirtualData
# np.save(fname + ".npy", self.data)
# dmap = open_memmap(fname + ".npy", mode="r")
# vdata = VirtualData([dmap, dmap])
# dummy = AnalogData(vdata, samplerate=1000)
# dummy.save(fname, overwrite=True)
# dummy2 = AnalogData(filename)
# assert dummy2.mode == "r+"
# assert np.array_equal(dummy2.data, vdata[:, :])
# del dummy, dummy2 # avoid PermissionError in Windows
# ensure trialdefinition is saved and loaded correctly
dummy = AnalogData(data=self.data,
trialdefinition=self.trl,
samplerate=1000)
dummy.save(fname + "_trl")
filename = construct_spy_filename(fname + "_trl", dummy)
dummy2 = load(filename)
assert np.array_equal(dummy.trialdefinition,
dummy2.trialdefinition)
# test getters
assert np.array_equal(dummy.sampleinfo, dummy2.sampleinfo)
assert np.array_equal(dummy._t0, dummy2._t0)
assert np.array_equal(dummy.trialinfo, dummy2.trialinfo)
del dummy, dummy2 # avoid PermissionError in Windows
# swap dimensions and ensure `dimord` is preserved
dummy = AnalogData(data=self.data,
dimord=["channel", "time"],
samplerate=1000)
dummy.save(fname + "_dimswap")
filename = construct_spy_filename(fname + "_dimswap", dummy)
dummy2 = load(filename)
assert dummy2.dimord == dummy.dimord
assert dummy2.channel.size == self.ns # swapped
assert dummy2.data.shape == dummy.data.shape
# Delete all open references to file objects and wait 0.1s for changes
# to take effect (thanks, Windows!)
del dummy, dummy2
time.sleep(0.1)