1 files changed, 36 insertions, 32 deletions

diff --git a/tests/test_repo_trace.py b/tests/test_repo_trace.py
index 5faf2938..e4aeb5de 100644
--- a/tests/test_repo_trace.py
+++ b/tests/test_repo_trace.py
@@ -22,35 +22,39 @@ import repo_trace
 
 
 class TraceTests(unittest.TestCase):
-  """Check Trace behavior."""
-
-  def testTrace_MaxSizeEnforced(self):
-    content = 'git chicken'
-
-    with repo_trace.Trace(content, first_trace=True):
-      pass
-    first_trace_size = os.path.getsize(repo_trace._TRACE_FILE)
-
-    with repo_trace.Trace(content):
-      pass
-    self.assertGreater(
-        os.path.getsize(repo_trace._TRACE_FILE), first_trace_size)
-
-    # Check we clear everything is the last chunk is larger than _MAX_SIZE.
-    with mock.patch('repo_trace._MAX_SIZE', 0):
-      with repo_trace.Trace(content, first_trace=True):
-        pass
-      self.assertEqual(first_trace_size,
-                       os.path.getsize(repo_trace._TRACE_FILE))
-
-      # Check we only clear the chunks we need to.
-      repo_trace._MAX_SIZE = (first_trace_size + 1) / (1024 * 1024)
-      with repo_trace.Trace(content, first_trace=True):
-        pass
-      self.assertEqual(first_trace_size * 2,
-                       os.path.getsize(repo_trace._TRACE_FILE))
-
-      with repo_trace.Trace(content, first_trace=True):
-        pass
-      self.assertEqual(first_trace_size * 2,
-                       os.path.getsize(repo_trace._TRACE_FILE))
+    """Check Trace behavior."""
+
+    def testTrace_MaxSizeEnforced(self):
+        content = "git chicken"
+
+        with repo_trace.Trace(content, first_trace=True):
+            pass
+        first_trace_size = os.path.getsize(repo_trace._TRACE_FILE)
+
+        with repo_trace.Trace(content):
+            pass
+        self.assertGreater(
+            os.path.getsize(repo_trace._TRACE_FILE), first_trace_size
+        )
+
+        # Check we clear everything is the last chunk is larger than _MAX_SIZE.
+        with mock.patch("repo_trace._MAX_SIZE", 0):
+            with repo_trace.Trace(content, first_trace=True):
+                pass
+            self.assertEqual(
+                first_trace_size, os.path.getsize(repo_trace._TRACE_FILE)
+            )
+
+            # Check we only clear the chunks we need to.
+            repo_trace._MAX_SIZE = (first_trace_size + 1) / (1024 * 1024)
+            with repo_trace.Trace(content, first_trace=True):
+                pass
+            self.assertEqual(
+                first_trace_size * 2, os.path.getsize(repo_trace._TRACE_FILE)
+            )
+
+            with repo_trace.Trace(content, first_trace=True):
+                pass
+            self.assertEqual(
+                first_trace_size * 2, os.path.getsize(repo_trace._TRACE_FILE)
+            )