def test_brunt_vaisala_frequency():
"""Test Brunt-Vaisala frequency function."""
truth = [[0.01163031, 0.01424416, 0.01744547, 0.01300308],
[0.01159041, 0.01417105, 0.01002045, 0.01003759],
[0.01155092, 0.00814010, 0., 0.01157062],
[0.01151183, np.nan, 0., 0.01628019]] * units('s^-1')
bv_freq = brunt_vaisala_frequency(bv_data()[0], bv_data()[1])
assert_almost_equal(bv_freq, truth, 6)
def test_brunt_vaisala_frequency():
"""Test Brunt-Vaisala frequency function."""
truth = [[0.01163031, 0.01424416, 0.01744547, 0.01300308],
[0.01159041, 0.01417105, 0.01002045, 0.01003759],
[0.01155092, 0.00814010, 0., 0.01157062],
[0.01151183, np.nan, 0., 0.01628019]] * units('s^-1')
bv_freq = brunt_vaisala_frequency(bv_data()[0], bv_data()[1])
assert_almost_equal(bv_freq, truth, 6)