PyIRoGlass v0.2.0 released!

sarahshi · Sep 24, 2023 · ec0decf · ec0decf
1 parent 1db290a
commit ec0decf
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/UnitTests/test_plotting.py b/UnitTests/test_plotting.py
@@ -5,7 +5,9 @@
 import pandas as pd
 import sys
 import PyIRoGlass as pig
-import matplotlib.pyplot as plt
+import matplotlib
+import matplotlib.pyplot
+from matplotlib import pyplot as plt
 
 
 class test_plotting_trace(unittest.TestCase):
@@ -66,7 +68,7 @@ def setUp(self):
         self.decimalPlace = 5
 
 
-    @patch.object(matplotlib.pyplot.Figure, 'savefig')
+    @patch('matplotlib.pyplot.Figure.savefig')
     def test_modelfit_savefile(self, mock_savefig):
 
         file_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '../src/PyIRoGlass/BaselineAvgPC.npz') 
@@ -90,7 +92,7 @@ def test_modelfit_savefile(self, mock_savefig):
         self.assertIn('dummy_filename.png', args[0])
         plt.close('all')
 
-    @patch.object(matplotlib.pyplot.Figure, 'savefig')
+    @patch('matplotlib.pyplot.Figure.savefig')
     def test_modelfit_nosavefile(self, mock_savefig):
 
         file_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '../src/PyIRoGlass/BaselineAvgPC.npz')