Ejemplo n.º 1
0
def test_normal_component(test_cross_xy):
    """Test getting cross section normal component of a 2D vector field."""
    norm_wind = normal_component(test_cross_xy['u_wind'],
                                 test_cross_xy['v_wind'])
    true_norm_wind_values = np.array(
        [[[
            32.21218429, 30.45650997, 28.59536112, 26.62832466, 24.57166983,
            22.43805311, 20.2347284
        ],
          [
              19.26516594, 17.41404337, 15.46613157, 13.42554447, 11.30508283,
              9.11378585, 6.85576955
          ],
          [
              6.3181476, 4.37157677, 2.33690202, 0.22276428, -1.96150417,
              -4.2104814, -6.52318931
          ],
          [
              -6.62887075, -8.67088982, -10.79232752, -12.98001592,
              -15.22809117, -17.53474865, -19.90214816
          ],
          [
              -19.5758891, -21.71335642, -23.92155707, -26.18279611,
              -28.49467817, -30.8590159, -33.28110701
          ]]])
    true_norm_wind = xr.DataArray(true_norm_wind_values * units('m/s'),
                                  coords=test_cross_xy['u_wind'].coords,
                                  dims=test_cross_xy['u_wind'].dims,
                                  attrs=test_cross_xy['u_wind'].attrs)
    assert_xarray_allclose(norm_wind, true_norm_wind)
Ejemplo n.º 2
0
def test_normal_component(test_cross_xy):
    """Test getting cross section normal component of a 2D vector field."""
    norm_wind = normal_component(test_cross_xy['u_wind'], test_cross_xy['v_wind'])
    true_norm_wind_values = np.array([[[32.21218429, 30.45650997, 28.59536112, 26.62832466,
                                        24.57166983, 22.43805311, 20.2347284],
                                       [19.26516594, 17.41404337, 15.46613157, 13.42554447,
                                        11.30508283, 9.11378585, 6.85576955],
                                       [6.3181476, 4.37157677, 2.33690202, 0.22276428,
                                        -1.96150417, -4.2104814, -6.52318931],
                                       [-6.62887075, -8.67088982, -10.79232752, -12.98001592,
                                        -15.22809117, -17.53474865, -19.90214816],
                                       [-19.5758891, -21.71335642, -23.92155707, -26.18279611,
                                        -28.49467817, -30.8590159, -33.28110701]]])
    true_norm_wind = xr.DataArray(true_norm_wind_values,
                                  coords=test_cross_xy['u_wind'].coords,
                                  dims=test_cross_xy['u_wind'].dims,
                                  attrs=test_cross_xy['u_wind'].attrs)
    assert_xarray_allclose(norm_wind, true_norm_wind)