Unidata / MetPy

    assert_almost_equal(motion.flatten(), truth, 8)

def test_bulk_shear():

    """Test bulk shear with observed sounding."""

    with UseSampleData():

        data = get_upper_air_data(datetime(2016, 5, 22, 0), 'DDC', source='wyoming')

    u, v = bulk_shear(data.variables['pressure'][:], data.variables['u_wind'][:],

                      data.variables['v_wind'][:], heights=data.variables['height'][:],

                      depth=6000 * units('meter'))

    truth = [29.899581266946115, -14.389225800205509] * units('knots')

    assert_almost_equal(u.to('knots'), truth[0], 8)

    assert_almost_equal(v.to('knots'), truth[1], 8)

def test_bulk_shear_no_depth():

    assert_almost_equal(v, 7.966031839967931 * units('m/s'), 7)

def test_mean_pressure_weighted_elevated():

    """Test pressure-weighted mean wind function with a base above the surface."""

    with UseSampleData():

        data = get_upper_air_data(datetime(2016, 5, 22, 0), 'DDC', source='wyoming')

    u, v = mean_pressure_weighted(data.variables['pressure'][:],

                                  data.variables['u_wind'][:],

                                  data.variables['v_wind'][:],

                                  heights=data.variables['height'][:],

                                  depth=3000 * units('meter'),

                                  bottom=data.variables['height'][0] + 3000 * units('meter'))

    assert_almost_equal(u, 8.270829843626476 * units('m/s'), 7)

    assert_almost_equal(v, 1.7392601775853547 * units('m/s'), 7)

def test_bunkers_motion():

    assert_almost_equal(v.to('knots'), truth[1], 8)

def test_bulk_shear_no_depth():

    """Test bulk shear with observed sounding and no depth given. Issue #568."""

    with UseSampleData():

        data = get_upper_air_data(datetime(2016, 5, 22, 0), 'DDC', source='wyoming')

    u, v = bulk_shear(data.variables['pressure'][:], data.variables['u_wind'][:],

                      data.variables['v_wind'][:], heights=data.variables['height'][:])

    truth = [20.225018939, 22.602359692] * units('knots')

    assert_almost_equal(u.to('knots'), truth[0], 8)

    assert_almost_equal(v.to('knots'), truth[1], 8)

def test_bulk_shear_elevated():