plot_utils: move pre_plot_setup and post_plot_setup to plot_utils
I need it for plot_allfreqs() which is not an object inheriting from
ThermalBase
diff --git a/tests/test_plot_utils.py b/tests/test_plot_utils.py
index ae83402..8deab38 100644
--- a/tests/test_plot_utils.py
+++ b/tests/test_plot_utils.py
@@ -16,3 +16,12 @@
"""Test normalize_title"""
self.assertEquals(plot_utils.normalize_title("Foo", ""), "Foo")
self.assertEquals(plot_utils.normalize_title("Foo", "Bar"), "Bar - Foo")
+
+ def test_post_plot_setup(self):
+ """Test that post_plot_setup() doesn't bomb"""
+
+ ax = plot_utils.pre_plot_setup()
+
+ plot_utils.post_plot_setup(ax)
+ plot_utils.post_plot_setup(ax, title="Foo")
+ plot_utils.post_plot_setup(ax, ylim=(0, 72))