test_lund_rescaling.py 文件源码

def test_lund_rescale_mean_velocity_different_grid():
    fileName = path.join(eddylicious.__path__[0], "..", "tests", "datasets",
                         "channel_flow_180", "dns.dat")
    dns = np.genfromtxt(fileName)

    eta = dns[:, 0]
    yPlus = dns[:, 1]
    u = dns[:, 1]
    v = np.zeros(u.shape)
    etaInfl = np.linspace(0, eta[-1], 119)
    yPlusInfl = etaInfl*6.37309e-2/3.5e-4
    uTest = interp1d(eta, u)(etaInfl)
    w = blending_function(etaInfl)
    uRescaled = lund_rescale_mean_velocity(eta, yPlus, u, v, etaInfl.size,
                                           etaInfl, yPlusInfl, 1,
                                           u[-1], u[-1], 1,
                                           w)[0][:, 0]
    for i in range(len(uTest)):
        assert_almost_equal(uTest[i], uRescaled[i], decimal=3)


# Test recaling dns data onto itself, using a different grid with eta > 1