test_radialV: clean up diag plot

MPoL-dev · Nov 2, 2023 · 55029c7 · 55029c7
1 parent afad0a9
commit 55029c7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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")