adding node attribute to vertex graphing test for code coverage

USEPA · Nov 13, 2023 · 7b8817c · 7b8817c
1 parent f2408d9
commit 7b8817c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/wntr/tests/test_graphics.py b/wntr/tests/test_graphics.py
@@ -113,7 +113,8 @@ def test_plot_network_vertices(self):
         wn = wntr.network.WaterNetworkModel(inp_file)
 
         wntr.graphics.plot_network(
-            wn, link_attribute="diameter", plot_vertices=True
+            wn, link_attribute="diameter", node_attribute="elevation",
+            plot_vertices=True
         )
         plt.savefig(filename, format="png")
         plt.close()