Corrected errors in pytest

diegopetrola · diegopetrola · commit ae2a558f4cd9 · 2020-08-13T18:57:02.000-03:00
Made it flake8 compliant
diff --git a/lib/matplotlib/font_manager.py b/lib/matplotlib/font_manager.py
@@ -695,6 +695,7 @@ def __init__(self,
         self._stretch = rcParams['font.stretch']
         self._size = rcParams['font.size']
         self._file = None
+        self._mathtext_fontset = rcParams['mathtext.fontset']
 
         if isinstance(family, str):
             # Treat family as a fontconfig pattern if it is the only
@@ -822,7 +823,7 @@ def get_fontconfig_pattern(self):
     
     def get_mathtext_fontset(self):
         """
-        Return the mathtext_fontset being used for this object.
+        Returns the mathtext_fontset being used for this object.
         
         See Also
         --------
@@ -947,7 +948,12 @@ def set_mathtext_fontset(self, fontset):
         """
         The mathtext_fontset is the family of fonts used in case this text is
         rendered in math mode.
-        
+
+        Parameters
+        ----------
+        fontset : str, the name of the fontset. Available fontsets are defined
+        in the matplotlibrc.template file
+
         See Also
         --------
         .text.set_mathtext_fontset
@@ -958,7 +964,7 @@ def set_mathtext_fontset(self, fontset):
         #this function validates the parameter fontset just as if it were
         #passed to rcParams['mathtext.fontset']
         _validators['mathtext.fontset'](fontset)
-        self._mathtext_fontset = fontset       
+        self._mathtext_fontset = fontset
 
     def copy(self):
         """Return a copy of self."""
diff --git a/lib/matplotlib/tests/test_mathtext.py b/lib/matplotlib/tests/test_mathtext.py
@@ -367,56 +367,55 @@ def test_mathtext_to_png(tmpdir):
 
 
 def test_mathtext_fontset():
-    #This is my first time doing this kinds of tests, so I am not sure if I am 
+    #This is my first time doing this kinds of tests, so I am not sure if I am
     #doing it the right way, here's my plan for this testing:
     #
-    #1) Create two figures, one without my implementation 
-    # (should produce 2 texts with the same font) 
+    #1) Create two figures, one without my implementation
+    # (should produce 2 texts with the same font)
     # and one with (should create two texts with different fonts)
     #2) Save the figures on Disk
     #3) Load the figures and compare them, if they are equal, the test fails
-    #4) Delete the figures 
+    #4) Delete the figures
     #
-    # I am making the test a bit slower but avoiding the burden of adding 
+    # I am making the test a bit slower but avoiding the burden of adding
     # more figures to the repo. On my computer at least, the time for this
     # test was of the order of a few seconds.
-    
+
     plt.rcParams["mathtext.fontset"] = "dejavusans"
 
     #figure 1 setup
-    #this figure IS NOT using the 'fontset' argument, therefore, 
+    #this figure IS NOT using the 'fontset' argument, therefore,
     #the font should be equal to the font set in rcParams["mathtext.fontset"]
-    f1, ax1 = plt.subplots(figsize=(12,10))
-    ax1.plot(range(11),color="0.9")
-    ax1.text(0.2,9,r"$\mathit{This\ text\ should\ have\ a\ one\ font} $",
+    f1, ax1 = plt.subplots(figsize=(12, 10))
+    ax1.plot(range(11), color="0.9")
+    ax1.text(0.2, 9, r"$\mathit{This\ text\ should\ have\ a\ one\ font} $",
              size=18)
-    
-    ax1.text(0.2,8,r"$\mathit{This\ text\ should\ have\ another} $",
+
+    ax1.text(0.2, 8, r"$\mathit{This\ text\ should\ have\ another} $",
              size=18)
-    
+
     #figure 2 setup
     #this figure IS using the 'fontset' parameter and must have two texts
     #with distinct fonts in the final image
-    f2, ax2 = plt.subplots(figsize=(12,10))
-    ax2.plot(range(11),color="0.9")
-    ax2.text(0.2,9,r"$\mathit{This\ text\ should\ have\ a\ one\ font} $",
-             size=18, fontset = 'dejavusans' )
-    
-    ax2.text(0.2,8,r"$\mathit{This\ text\ should\ have\ another} $",
-             size = 18, fontset = 'dejavuserif' )
-    
+    f2, ax2 = plt.subplots(figsize=(12, 10))
+    ax2.plot(range(11), color="0.9")
+    ax2.text(0.2, 9, r"$\mathit{This\ text\ should\ have\ a\ one\ font} $",
+             size=18, fontset='dejavusans')
+
+    ax2.text(0.2, 8, r"$\mathit{This\ text\ should\ have\ another} $",
+             size=18, fontset='dejavuserif')
+
     f1.savefig('mathtext_fontset_image1.png')
     f2.savefig('mathtext_fontset_image2.png')
-    
+
     figureLoaded1 = io.FileIO('mathtext_fontset_image1.png').readall()
     figureLoaded2 = io.FileIO('mathtext_fontset_image2.png').readall()
-    
+
     figuresEqual = figureLoaded1 == figureLoaded2
-    
+
     os.remove('mathtext_fontset_image1.png')
     os.remove('mathtext_fontset_image2.png')
-    
+
     #if the parameter fontset is not working, the figures will be equal and the
     #test will fail
     assert not figuresEqual
-    
diff --git a/lib/matplotlib/text.py b/lib/matplotlib/text.py
@@ -1044,19 +1044,23 @@ def set_fontsize(self, fontsize):
         """
         self._fontproperties.set_size(fontsize)
         self.stale = True
-        
+
     def set_mathtext_fontset(self, fontset):
         """
-        The mathtext_fontset will be the fontset used in case this text 
+        The mathtext_fontset will be the fontset used in case this text
         is rendered as mathtext
 
-        The purpouse of the function is be simmilar to rcParams['mathtext.fontset']
+        The purpouse of the function is be simmilar to rcParams
         but for each Text element instead of global.
-        
+
         Parameters
         ----------
         fontset : str, the name of the fontset. Available fontsets are defined
         in the matplotlibrc.template file
+
+        See also
+        ----------
+        matplotlib.rcParams
         """
         self._fontproperties.set_mathtext_fontset(fontset)
 
