def test_avg_mag_dir_nan():
num_bins = 30
num_beams = 4
vel = EarthVelocity(num_bins, num_beams)
vel.Magnitude = [np.NaN, np.NaN]
vel.Direction = [np.NaN, np.NaN]
avg_mag, avg_dir = vel.average_mag_dir()
assert avg_mag is None
assert avg_dir is None
def test_avg_mag_dir_nan():
num_bins = 30
num_beams = 4
vel = EarthVelocity(num_bins, num_beams)
vel.Magnitude = [5.0, 5.0, 5.0, 5.0, np.NaN]
vel.Direction = [1.2, 1.2, 1.2, 1.2, np.NaN]
avg_mag, avg_dir = vel.average_mag_dir()
assert avg_mag == pytest.approx(5, 0.001)
assert avg_dir == pytest.approx(1.2, 0.001)