2d network image tests

Author: wx4stg

tests/test_base_plot.py

@@ -3,6 +3,7 @@
 from pyxlma.plot.xlma_base_plot import *
 from pyxlma.plot.xlma_plot_feature import *
 from pyxlma.lmalib.grid import *
+from pyxlma.lmalib.io import read as lma_read
 import datetime as dt
 import pandas as pd
 import matplotlib.dates as md
@@ -135,4 +136,33 @@ def test_plot_feature_inset_view():
                 c=colors, edgecolors='k', linewidths=0.1, add_to_histogram=True)
     inset_view(bk_plot, dataset.event_longitude.data, dataset.event_latitude.data,
                [-102.75, -102.25], [32, 32.5], .01, .01)
-    return bk_plot.fig
+    return bk_plot.fig
+
+
+@pytest.mark.mpl_image_compare
+def test_plot_feature_ndln_time():
+    start_time = dt.datetime(2023, 12, 24, 0, 57, 0, 0)
+    end_time = start_time + dt.timedelta(seconds=60)
+    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')
+    nldn_data = lma_read.nldn('examples/network_samples/gld360enldnns_20231224_daily_v1_lit.raw')
+    plot_2d_network_points(bk_plot, nldn_data)
+    return bk_plot.fig
+
+
+@pytest.mark.mpl_image_compare
+def test_plot_feature_entln_real_height():
+    start_time = dt.datetime(2023, 12, 24, 0, 57, 0, 0)
+    end_time = start_time + dt.timedelta(seconds=60)
+    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')
+    entln_data = lma_read.entln('examples/network_samples/lxarchive_pulse20231224.csv')
+    plot_2d_network_points(bk_plot, entln_data, actual_height=entln_data['icheight'])
+    return bk_plot.fig
+
+@pytest.mark.mpl_image_compare
+def test_plot_feature_ndln_polarity():
+    start_time = dt.datetime(2023, 12, 24, 0, 57, 0, 0)
+    end_time = start_time + dt.timedelta(seconds=60)
+    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')
+    nldn_data = lma_read.nldn('examples/network_samples/gld360enldnns_20231224_daily_v1_lit.raw')
+    plot_2d_network_points(bk_plot, nldn_data, color_by='polarity')
+    return bk_plot.fig

tests/test_io.py

@@ -25,6 +25,7 @@ def test_read_onefile_dataset():
     for var in truth.data_vars:
         compare_dataarrays(dataset, truth, var)
 
+
 def test_read_nldn():
     dataset = lma_read.nldn('examples/network_samples/gld360enldnns_20231224_daily_v1_lit.raw')
     truth = pd.DataFrame({
@@ -43,6 +44,7 @@ def test_read_nldn():
     })
     assert dataset.equals(truth)
 
+
 def test_read_entln():
     dataset = lma_read.entln('examples/network_samples/lxarchive_pulse20231224.csv')
     truth = pd.DataFrame({