Fix tests montage unit.

jackz314 · Jun 15, 2023 · 3826196 · 3826196
1 parent ed17572
commit 3826196
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/eeglabio/tests/test_epochs.py b/eeglabio/tests/test_epochs.py
@@ -39,7 +39,7 @@ def test_export_set(tmpdir, preload):
     # epochs.set_annotations(annot)
     temp_fname = op.join(str(tmpdir), 'test_epochs.set')
     export_mne_epochs(epochs, temp_fname)
-    epochs_read = read_epochs_eeglab(temp_fname)
+    epochs_read = read_epochs_eeglab(temp_fname, montage_units='m')
     assert epochs.ch_names == epochs_read.ch_names
     cart_coords = np.array([d['loc'][:3]
                             for d in epochs.info['chs']])  # just xyz

diff --git a/eeglabio/tests/test_raw.py b/eeglabio/tests/test_raw.py
@@ -16,7 +16,7 @@ def test_export_set(tmpdir):
         meg=True, eeg=True, ecg=True).load_data()
     temp_fname = op.join(str(tmpdir), 'test_raw.set')
     export_mne_raw(raw, temp_fname)
-    raw_read = read_raw_eeglab(temp_fname, preload=True)
+    raw_read = read_raw_eeglab(temp_fname, preload=True, montage_units='m')
     assert raw.ch_names == raw_read.ch_names
     cart_coords = np.array([d['loc'][:3] for d in raw.info['chs']])  # just xyz
     cart_coords_read = np.array([d['loc'][:3] for d in raw_read.info['chs']])