import numpy as np
import mne

for ch_type in ('eeg', 'seeg', 'ecog', 'dbs'):
    raw = mne.io.RawArray(
        np.random.random((10, 1000)),
        mne.create_info([f'ch{i}' for i in range(1, 11)],  1000, [ch_type] * 10))
    montage = mne.channels.make_dig_montage(
        {ch: np.random.random((3,)) for ch in raw.ch_names}, lpa=[1, 0, 0], nasion=[0, 0.1, 0], rpa=[-1, 0, 0],
        coord_frame='mri')
    raw.set_montage(montage)
    raw.save('tmp-raw.fif', overwrite=True, verbose=False)
    print(ch_type)
    print('Before: ', raw.info['chs'][0]['coord_frame'])
    info = mne.io.read_info('tmp-raw.fif')
    print('After : ', info['chs'][0]['coord_frame'])