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)