def test_wera_qc():
radial_file = data_path / "radials" / "ruv" / "WERA" / "RDL_csw_2019_10_24_162300.ruv"
rad1 = Radial(radial_file, replace_invalid=False)
rad1.initialize_qc()
# assert len(rad1.data) == 6327
# rad1.mask_over_land(subset=True)
rad1.qc_qartod_radial_count()
rad1.qc_qartod_valid_location()
rad1.qc_qartod_maximum_velocity()
rad1.qc_qartod_spatial_median()
rad1.qc_qartod_avg_radial_bearing(reference_bearing=180)
rad1.qc_qartod_primary_flag()
# assert len(rad1.data) == 5745
assert "QC07" in rad1.data
assert "QC08" not in rad1.data # no VFLG column so we can't run it
assert "QC09" in rad1.data
assert "QC10" in rad1.data
# assert 'QC11' in rad1.data # temporal gradient test
assert "QC12" in rad1.data
assert "PRIM" in rad1.data
def test_codar_qc():
radial_file = data_path / "radials" / "ruv" / "SEAB" / "RDLi_SEAB_2019_01_01_0100.ruv"
radial_file_previous = data_path / "radials" / "ruv" / "SEAB" / "RDLi_SEAB_2019_01_01_0000.ruv"
rad1 = Radial(radial_file, replace_invalid=False)
rad1.initialize_qc()
# assert len(rad1.data) == 733
# rad1.mask_over_land(subset=True)
rad1.qc_qartod_radial_count()
rad1.qc_qartod_valid_location()
rad1.qc_qartod_maximum_velocity()
rad1.qc_qartod_spatial_median()
rad1.qc_qartod_temporal_gradient(radial_file_previous)
rad1.qc_qartod_avg_radial_bearing(reference_bearing=180)
rad1.qc_qartod_primary_flag()
# assert len(rad1.data) == 587
assert "QC07" in rad1.data
assert "QC08" in rad1.data
assert "QC09" in rad1.data
assert "QC10" in rad1.data
assert "QC11" in rad1.data # temporal gradient test
assert "QC12" in rad1.data
assert "PRIM" in rad1.data