def test_vertical_integral(test_dataset): ds = test_dataset H = ds.attrs['H'] dz = ds.attrs['dz'] f = np.sin(np.pi * ds.Z.values / H) ds['f'] = (('Z'), f) ds['fint'] = (f*dz).sum() ds['favg'] = ds['fint'] / H gcm = GCMDataset(ds) gcm_fint = gcm.integrate_z(ds.f) assert gcm_fint.equals(ds.fint), (gcm_fint, ds.fint) gcm_favg = gcm.integrate_z(ds.f, average=True) assert gcm_favg.equals(ds.favg), (gcm_favg, ds.favg)
def test_vertical_integral(test_dataset): ds = test_dataset H = ds.attrs["H"] dz = ds.attrs["dz"] f = np.sin(np.pi * ds.Z.values / H) ds["f"] = (("Z"), f) ds["fint"] = (f * dz).sum() ds["favg"] = ds["fint"] / H gcm = GCMDataset(ds) gcm_fint = gcm.integrate_z(ds.f) assert gcm_fint.equals(ds.fint), (gcm_fint, ds.fint) gcm_favg = gcm.integrate_z(ds.f, average=True) assert gcm_favg.equals(ds.favg), (gcm_favg, ds.favg)
def test_vertical_integral(test_dataset): ds = test_dataset H = ds.attrs['H'] dz = ds.attrs['dz'] f = np.sin(np.pi * ds.Z.values / H) ds['f'] = (('Z'), f) ds['fint'] = (f * dz).sum() ds['favg'] = ds['fint'] / H gcm = GCMDataset(ds) gcm_fint = gcm.integrate_z(ds.f) assert gcm_fint.equals(ds.fint), (gcm_fint, ds.fint) gcm_favg = gcm.integrate_z(ds.f, average=True) assert gcm_favg.equals(ds.favg), (gcm_favg, ds.favg)