def test_find_cluster():
t = Target(toiid=837)
_ = t.query_gaia_dr2_catalog(return_nearest_xmatch=True)
nearest_cluster_member = t.get_nearest_cluster_member(
catalog_name="CantatGaudin2020", match_id=True)
assert isinstance(nearest_cluster_member, pd.Series)
assert t.nearest_cluster_name == "IC_2602"
def test_target_xmatch():
t = Target(toiid=837)
_ = t.query_gaia_dr2_catalog(return_nearest_xmatch=True)
_ = t.query_tic_catalog(return_nearest_xmatch=True)
assert t.validate_gaia_tic_xmatch()
def test_query_gaia():
t = Target(gaiaDR2id=5251470948229949568)
gaia_params1 = t.query_gaia_dr2_catalog(radius=60)
assert isinstance(gaia_params1, pd.DataFrame)
gaia_params2 = t.query_gaia_dr2_catalog(return_nearest_xmatch=True)
assert isinstance(gaia_params2, pd.Series)
def plot_cluster_kinematics(
ticid=None,
toiid=None,
cluster_name=None,
frac_err=0.5,
rv=None,
savefig=False,
):
"""
"""
assert (ticid is not None) | (toiid is not None)
t = Target(toiid=toiid, ticid=ticid)
if cluster_name is None:
cluster, idxs = t.get_cluster_membership(
catalog_name="CantatGaudin2020",
return_idxs=True,
frac_err=frac_err,
sigma=5,
)
cluster_name = cluster.Cluster
c = Cluster(cluster_name=cluster_name)
df_target = t.query_gaia_dr2_catalog(return_nearest_xmatch=True)
if rv is not None:
df_target.radial_velocity = rv
else:
if np.isnan(df_target.radial_velocity):
rv = np.nanmean(list(t.query_vizier_param("RV").values()))
if not np.isnan(rv):
df_target.radial_velocity = rv
try:
fig1 = c.plot_xyz_uvw(target_gaiaid=t.gaiaid,
df_target=df_target,
match_id=False)
fig1.suptitle(f"{t.target_name} in {c.cluster_name}")
if savefig:
fp1 = f"{t.target_name}_galactocentric.png"
fig1.savefig(fp1, bbox_inches="tight")
except Exception as e:
print("Error: ", e)
# ==============
try:
log10age = c.get_cluster_age()
fig2 = c.plot_rdp_pmrv(target_gaiaid=t.gaiaid,
df_target=df_target,
match_id=False)
fig2.suptitle(f"{t.target_name} in {c.cluster_name}")
if savefig:
fp2 = f"{t.target_name}_kinematics.png"
fig2.savefig(fp2, bbox_inches="tight")
except Exception as e:
print("Error: ", e)
# ==============
try:
# TODO: AG50 doesn't yield G consistent with cmd
# if str(df_target.a_g_val) == "nan":
# vq = t.query_vizier_param("AG50")
# if "I/349/starhorse" in vq:
# df_target.a_g_val = vq["I/349/starhorse"]
# print("Using AG from starhorse.")
log10age = c.get_cluster_age()
ax = c.plot_cmd(
target_gaiaid=t.gaiaid,
df_target=df_target,
match_id=False,
log_age=log10age,
)
ax.set_title(f"{t.target_name} in {c.cluster_name}")
if savefig:
fp3 = f"{t.target_name}_cmd.png"
ax.figure.savefig(fp3, bbox_inches="tight")
except Exception as e:
print("Error: ", e)
try:
ax = c.plot_hrd(
target_gaiaid=t.gaiaid,
df_target=df_target,
match_id=False,
log_age=log10age,
)
ax.set_title(f"{t.target_name} in {c.cluster_name}")
if savefig:
fp4 = f"{t.target_name}_hrd.png"
ax.figure.savefig(fp4, bbox_inches="tight")
except Exception as e:
print("Error: ", e)