test_xr_accessor.py 文件源码

def test_set_master_clock_dim(self):
        ds = xr.Dataset(coords={'clock': [1, 2], 'clock2': [3, 4]})

        ds.xsimlab._set_master_clock_dim('clock')
        assert self._master_clock_key in ds.clock.attrs

        ds.xsimlab._set_master_clock_dim('clock2')
        assert self._master_clock_key not in ds.clock.attrs
        assert self._master_clock_key in ds.clock2.attrs

        with pytest.raises(KeyError):
            ds.xsimlab._set_master_clock_dim('invalid_clock')