mirror of
https://github.com/pim-n/pg-rad
synced 2026-03-23 21:58:12 +01:00
add plotting to test case. Add saveplot argument. Improve logging in LandscapeSlicePlotter
This commit is contained in:
@ -36,10 +36,10 @@ class LandscapeSlicePlotter:
|
||||
ax.set_aspect("equal")
|
||||
|
||||
if save:
|
||||
name = landscape.name.lower().replace(' ', '_')
|
||||
plt.savefig(
|
||||
f"{name}_z{self.z}.png"
|
||||
)
|
||||
landscape_name = landscape.name.lower().replace(' ', '_')
|
||||
filename = f"{landscape_name}_z{self.z}.png"
|
||||
plt.savefig(filename)
|
||||
logger.info("Plot saved to file: "+filename)
|
||||
|
||||
if show:
|
||||
plt.show()
|
||||
|
||||
Reference in New Issue
Block a user