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')