Remove extra space and add context manager to test

Author: sakarpanta

pandas/core/frame.py

@@ -2497,12 +2497,12 @@ def memory_usage(self, index=True, deep=False):
         ...              for t in dtypes])
         >>> df = pd.DataFrame(data)
         >>> df.head()
-           int64  float64   complex128  object  bool
-        0      1      1.0     1.0+0.0j       1  True
-        1      1      1.0     1.0+0.0j       1  True
-        2      1      1.0     1.0+0.0j       1  True
-        3      1      1.0     1.0+0.0j       1  True
-        4      1      1.0     1.0+0.0j       1  True
+           int64  float64  complex128  object  bool
+        0      1      1.0    1.0+0.0j       1  True
+        1      1      1.0    1.0+0.0j       1  True
+        2      1      1.0    1.0+0.0j       1  True
+        3      1      1.0    1.0+0.0j       1  True
+        4      1      1.0    1.0+0.0j       1  True
 
         >>> df.memory_usage()
         Index            80

pandas/tests/io/formats/test_format.py

@@ -1371,16 +1371,16 @@ def test_to_string_float_index(self):
         assert result == expected
 
     def test_to_string_complex_float_formatting(self):
-        fmt.set_option('display.precision', 5)
-        df = DataFrame({'x': [
-            (0.4467846931321966 + 0.0715185102060818j),
-            (0.2739442392974528 + 0.23515228785438969j),
-            (0.26974928742135185 + 0.3250604054898979j)]})
-        result = df.to_string()
-        expected = ('                  x\n0  0.44678+0.07152j\n'
-                    '1  0.27394+0.23515j\n'
-                    '2  0.26975+0.32506j')
-        assert result == expected
+        with pd.option_context('display.precision', 5):
+            df = DataFrame({'x': [
+                (0.4467846931321966 + 0.0715185102060818j),
+                (0.2739442392974528 + 0.23515228785438969j),
+                (0.26974928742135185 + 0.3250604054898979j)]})
+            result = df.to_string()
+            expected = ('                  x\n0  0.44678+0.07152j\n'
+                        '1  0.27394+0.23515j\n'
+                        '2  0.26975+0.32506j')
+            assert result == expected
 
     def test_to_string_ascii_error(self):
         data = [('0  ', '                        .gitignore ', '     5 ',