def test_remove_nan_dia_sources(self):
"""Test removing DiaSources with NaN locations.
"""
self.diaSources.loc[2, "ra"] = np.nan
self.diaSources.loc[3, "decl"] = np.nan
self.diaSources.loc[4, "ra"] = np.nan
self.diaSources.loc[4, "decl"] = np.nan
assoc_task = AssociationTask()
out_dia_sources = assoc_task.check_dia_source_radec(self.diaSources)
self.assertEqual(len(out_dia_sources), len(self.diaSources) - 3)
def test_remove_nan_dia_sources(self):
n_sources = 6
dia_sources = create_test_points_pandas(
point_locs_deg=[
[0.04 * (src_idx + 1),
0.04 * (src_idx + 1)]
for src_idx in range(n_sources)],
start_id=0,
scatter_arcsec=-1)
dia_sources.rename(columns={"coord_ra": "ra",
"coord_dec": "decl",
"id": "diaSourceId"},
inplace=True)
dia_sources.loc[2, "ra"] = np.nan
dia_sources.loc[3, "decl"] = np.nan
dia_sources.loc[4, "ra"] = np.nan
dia_sources.loc[4, "decl"] = np.nan
assoc_task = AssociationTask()
out_dia_sources = assoc_task.check_dia_source_radec(dia_sources)
self.assertEqual(len(out_dia_sources), n_sources - 3)