Skip to content

Commit 775b992

Browse files
committed
update tests
1 parent a981438 commit 775b992

3 files changed

Lines changed: 10 additions & 1 deletion

File tree

tests/test_base_plot.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,5 +195,14 @@ def test_plot_feature_glm_events():
195195
end_time = start_time + dt.timedelta(seconds=60)
196196
bk_plot = BlankPlot(start_time, bkgmap=True, xlim=[-103.5, -99.5], ylim=[31.5, 35.5], zlim=[0, 20], tlim=[start_time, end_time], title='XLMA Test Plot')
197197
glm_data = GLMDataset('examples/network_samples/OR_GLM-L2-LCFA_G16_s20233580057000_e20233580057200_c20233580057222.nc')
198-
plot_glm_events(glm_data.dataset, bk_plot, fake_alt=[0, 0.25], poly_kwargs={'cmap' : 'plasma'}, vlines_kwargs={'linewidths' : 0.5})
198+
plot_glm_events(glm_data.dataset, bk_plot, fake_alt=[0, 0.25], poly_kwargs={'cmap' : 'plasma'}, vlines_kwargs={'linewidths' : 0.5}, should_parallax_correct=False)
199+
return bk_plot.fig
200+
201+
@pytest.mark.mpl_image_compare
202+
def test_plot_feature_glm_events_parallax():
203+
start_time = dt.datetime(2023, 12, 24, 0, 57, 0, 0)
204+
end_time = start_time + dt.timedelta(seconds=60)
205+
bk_plot = BlankPlot(start_time, bkgmap=True, xlim=[-103.5, -99.5], ylim=[31.5, 35.5], zlim=[0, 20], tlim=[start_time, end_time], title='XLMA Test Plot')
206+
glm_data = GLMDataset('examples/network_samples/OR_GLM-L2-LCFA_G16_s20233580057000_e20233580057200_c20233580057222.nc')
207+
plot_glm_events(glm_data.dataset, bk_plot, fake_alt=[0, 0.25], poly_kwargs={'cmap' : 'plasma'}, vlines_kwargs={'linewidths' : 0.5}, should_parallax_correct=True)
199208
return bk_plot.fig
-200 Bytes
Loading
100 KB
Loading

0 commit comments

Comments
 (0)