test_data.py 文件源码

def test_update_05_1variable(self):
        """Test to change the variable"""
        ds = psyd.open_dataset(bt.get_file('test-t2m-u-v.nc'))
        arr = ds.psy.t2m.psy[0, 0, 0]
        arr.attrs['test'] = 4
        self.assertNotIn('test', ds.t2m.attrs)
        self.assertIs(arr.psy.base, ds)
        self.assertEqual(dict(arr.psy.idims), {'time': 0, 'lev': 0, 'lat': 0,
                                               'lon': slice(None)})
        # update to next time step
        arr.psy.update(name='u', time=1)
        self.assertEqual(arr.time, ds.time[1])
        self.assertEqual(arr.name, 'u')
        self.assertEqual(arr.values.tolist(),
                         ds.u[1, 0, 0, :].values.tolist())
        self.assertEqual(dict(arr.psy.idims), {'time': 1, 'lev': 0, 'lat': 0,
                                               'lon': slice(None)})
        self.assertNotIn('test', ds.t2m.attrs)
        self.assertIn('test', arr.attrs)
        self.assertEqual(arr.test, 4)