test fig return

tests/adler/utilities/test_plotting_utilities.py

@@ -1,9 +1,9 @@
 import pytest
 import matplotlib
 
-from adler.dataclasses.AdlerPlanetoid import AdlerPlanetoid
+from adler.objectdata.AdlerPlanetoid import AdlerPlanetoid
 from adler.utilities.tests_utilities import get_test_data_filepath
-import adler.utilities.plotting_utilities as plot_utils
+from adler.utilities.plotting_utilities import plot_errorbar
 
 # set up test planetoid object
 ssoid = 8268570668335894776
@@ -14,5 +14,16 @@
 
 
 def test_plot_errorbar_return(planetoid=test_planetoid, filt_list=["r"]):
-    fig = plot_utils.plot_errorbar(planetoid, filt_list=filt_list)
-    assert type(fig) == matplotlib.figure.Figure
+
+    # make the fig object
+    fig = plot_errorbar(planetoid, filt_list=filt_list)
+    print(isinstance(fig, matplotlib.figure.Figure))
+
+    # check that fig is of the correct type
+    assert isinstance(fig, matplotlib.figure.Figure)
+
+
+if __name__ == "__main__":
+    print(test_planetoid.__dict__)
+
+    test_plot_errorbar_return()