def test_loading(self):
"""Loads data correctly."""
dat = load_mushu_data('foo.meta')
self.assertEqual(dat.fs, 1)
np.testing.assert_array_equal(dat.markers, [[0.0, 'Marker 0'], [500.0, 'Marker 500'], ['666.666', 'Marker 666.666']])
np.testing.assert_array_equal(dat.axes[0], [0, 1000, 2000])
np.testing.assert_array_equal(dat.axes[1], ['ch1', 'ch2'])
self.assertEqual(dat.names, ['time', 'channels'])
self.assertEqual(dat.units, ['ms', '#'])
data = np.array([0, 1, 33, 66, .1, .2], np.float32).reshape(-1, 2)
np.testing.assert_array_equal(dat.data, data)
def test_loading(self):
"""Loads data correctly."""
dat = load_mushu_data('foo.meta')
self.assertEqual(dat.fs, 1)
np.testing.assert_array_equal(dat.markers, [[0.0, 'Marker 0'], [500.0, 'Marker 500'], ['666.666', 'Marker 666.666']])
np.testing.assert_array_equal(dat.axes[0], [0, 1000, 2000])
np.testing.assert_array_equal(dat.axes[1], ['ch1', 'ch2'])
self.assertEqual(dat.names, ['time', 'channels'])
self.assertEqual(dat.units, ['ms', '#'])
data = np.array([0, 1, 33, 66, .1, .2], np.float32).reshape(-1, 2)
np.testing.assert_array_equal(dat.data, data)
def test_incomplete_fileset(self):
"""Must raise an error if not all files are available."""
os.remove('foo.marker')
with self.assertRaises(AssertionError):
load_mushu_data('foo.meta')
def test_incomplete_fileset(self):
"""Must raise an error if not all files are available."""
os.remove('foo.marker')
with self.assertRaises(AssertionError):
load_mushu_data('foo.meta')
