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)