Python GCMDataset.integrate_z 예제들

예제 #1

파일: test_gcm_dataset.py 프로젝트: pittwolfe/xgcm

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)

예제 #2

파일: test_gcm_dataset.py 프로젝트: zhaobin74/xgcm

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)

예제 #3

파일: test_gcm_dataset.py 프로젝트: stb2145/xgcm

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)