From 55029c7225f5aa28e4509b26a145a45d617246ee Mon Sep 17 00:00:00 2001 From: Jeff Jennings Date: Wed, 1 Nov 2023 20:42:05 -0400 Subject: [PATCH] test_radialV: clean up diag plot --- test/onedim_test.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/test/onedim_test.py b/test/onedim_test.py index abef8519..476c2144 100644 --- a/test/onedim_test.py +++ b/test/onedim_test.py @@ -77,11 +77,13 @@ def test_radialV(mock_1d_vis_model, tmp_path): fig, ax = plt.subplots() ax.plot(q_dep / 1e6, Vtrue_dep, 'k.', label='truth deprojected') - ax.plot(qtest / 1e6, Vtest, 'r.-', label='recovery') + ax.plot(qtest / 1e3, Vtest, 'r.-', label='recovery') + ax.set_xlim(-0.5, 6) ax.set_xlabel(r'Baseline [M$\lambda$]') ax.set_ylabel('Re(V) [Jy]') - ax.set_title(f"Geometry {geom}") + ax.set_title(f"Geometry {geom}", fontsize=10) + ax.legend() fig.savefig(tmp_path / "test_radialV.png", dpi=300) plt.close("all")