ran black on latex.py and test_to_latex.py (pandas-dev#25436)

Author: jeschwar

pandas/io/formats/latex.py

@@ -121,7 +121,7 @@ def pad_empties(x):
         else:
             self._write_tabular_begin(buf, column_format)
 
-        buf.write('\\toprule\n')
+        buf.write("\\toprule\n")
 
         ilevels = self.frame.index.nlevels
         clevels = self.frame.columns.nlevels
@@ -287,24 +287,21 @@ def _write_tabular_begin(self, buf, column_format):
         if self.caption is not None or self.label is not None:
             # then write output in a nested table/tabular environment
             if self.caption is None:
-                caption_ = ''
+                caption_ = ""
             else:
-                caption_ = '\n\\caption{{{}}}'.format(self.caption)
+                caption_ = "\n\\caption{{{}}}".format(self.caption)
 
             if self.label is None:
-                label_ = ''
+                label_ = ""
             else:
-                label_ = '\n\\label{{{}}}'.format(self.label)
+                label_ = "\n\\label{{{}}}".format(self.label)
 
-            buf.write('\\begin{{table}}\n\\centering{}{}\n'.format(
-                caption_,
-                label_
-            ))
+            buf.write("\\begin{{table}}\n\\centering{}{}\n".format(caption_, label_))
         else:
             # then write output only in a tabular environment
             pass
 
-        buf.write('\\begin{{tabular}}{{{fmt}}}\n'.format(fmt=column_format))
+        buf.write("\\begin{{tabular}}{{{fmt}}}\n".format(fmt=column_format))
 
     def _write_tabular_end(self, buf):
         """
@@ -318,10 +315,10 @@ def _write_tabular_end(self, buf):
             a string.
 
         """
-        buf.write('\\bottomrule\n')
-        buf.write('\\end{tabular}\n')
+        buf.write("\\bottomrule\n")
+        buf.write("\\end{tabular}\n")
         if self.caption is not None or self.label is not None:
-            buf.write('\\end{table}\n')
+            buf.write("\\end{table}\n")
         else:
             pass
 
@@ -341,23 +338,23 @@ def _write_longtable_begin(self, buf, column_format):
             for 3 columns
 
         """
-        buf.write('\\begin{{longtable}}{{{fmt}}}\n'.format(fmt=column_format))
+        buf.write("\\begin{{longtable}}{{{fmt}}}\n".format(fmt=column_format))
 
         if self.caption is not None or self.label is not None:
             if self.caption is None:
                 pass
             else:
-                buf.write('\\caption{{{}}}'.format(self.caption))
+                buf.write("\\caption{{{}}}".format(self.caption))
 
             if self.label is None:
                 pass
             else:
-                buf.write('\\label{{{}}}'.format(self.label))
+                buf.write("\\label{{{}}}".format(self.label))
 
             # a double-backslash is required at the end of the line
             # as discussed here:
             # https://tex.stackexchange.com/questions/219138
-            buf.write('\\\\\n')
+            buf.write("\\\\\n")
         else:
             pass
 
@@ -373,4 +370,4 @@ def _write_longtable_end(buf):
             a string.
 
         """
-        buf.write('\\end{longtable}\n')
+        buf.write("\\end{longtable}\n")

pandas/tests/io/formats/test_to_latex.py

@@ -440,15 +440,13 @@ def test_to_latex_longtable(self):
 
     def test_to_latex_caption_label(self):
         # GH 25436
-        the_caption = 'a table in a \\texttt{table/tabular} environment'
-        the_label = 'tab:table_tabular'
+        the_caption = "a table in a \\texttt{table/tabular} environment"
+        the_label = "tab:table_tabular"
 
-        df = DataFrame({'a': [1, 2], 'b': ['b1', 'b2']})
+        df = DataFrame({"a": [1, 2], "b": ["b1", "b2"]})
 
         # test when only the caption is provided
-        result_c = df.to_latex(
-            caption=the_caption
-        )
+        result_c = df.to_latex(caption=the_caption)
 
         expected_c = r"""\begin{table}
 \centering
@@ -466,9 +464,7 @@ def test_to_latex_caption_label(self):
         assert result_c == expected_c
 
         # test when only the label is provided
-        result_l = df.to_latex(
-            label=the_label
-        )
+        result_l = df.to_latex(label=the_label)
 
         expected_l = r"""\begin{table}
 \centering
@@ -486,10 +482,7 @@ def test_to_latex_caption_label(self):
         assert result_l == expected_l
 
         # test when the caption and the label are provided
-        result_cl = df.to_latex(
-            caption=the_caption,
-            label=the_label
-        )
+        result_cl = df.to_latex(caption=the_caption, label=the_label)
 
         expected_cl = r"""\begin{table}
 \centering
@@ -509,16 +502,13 @@ def test_to_latex_caption_label(self):
 
     def test_to_latex_longtable_caption_label(self):
         # GH 25436
-        the_caption = 'a table in a \\texttt{longtable} environment'
-        the_label = 'tab:longtable'
+        the_caption = "a table in a \\texttt{longtable} environment"
+        the_label = "tab:longtable"
 
-        df = DataFrame({'a': [1, 2], 'b': ['b1', 'b2']})
+        df = DataFrame({"a": [1, 2], "b": ["b1", "b2"]})
 
         # test when only the caption is provided
-        result_c = df.to_latex(
-            longtable=True,
-            caption=the_caption
-        )
+        result_c = df.to_latex(longtable=True, caption=the_caption)
 
         expected_c = r"""\begin{longtable}{lrl}
 \caption{a table in a \texttt{longtable} environment}\\
@@ -540,10 +530,7 @@ def test_to_latex_longtable_caption_label(self):
         assert result_c == expected_c
 
         # test when only the label is provided
-        result_l = df.to_latex(
-            longtable=True,
-            label=the_label,
-        )
+        result_l = df.to_latex(longtable=True, label=the_label)
 
         expected_l = r"""\begin{longtable}{lrl}
 \label{tab:longtable}\\
@@ -565,11 +552,7 @@ def test_to_latex_longtable_caption_label(self):
         assert result_l == expected_l
 
         # test when the caption and the label are provided
-        result_cl = df.to_latex(
-            longtable=True,
-            caption=the_caption,
-            label=the_label,
-        )
+        result_cl = df.to_latex(longtable=True, caption=the_caption, label=the_label)
 
         expected_cl = r"""\begin{longtable}{lrl}
 \caption{a table in a \texttt{longtable} environment}\label{tab:longtable}\\