Make Analysis a context manager to properly close sqlite filesnedbat/close-data

1 files changed, 6 insertions, 0 deletions

diff --git a/coverage/results.py b/coverage/results.py
index ba335209..83cdbdf9 100644
--- a/coverage/results.py
+++ b/coverage/results.py
@@ -49,6 +49,12 @@ class Analysis(object):
             n_missing_branches=n_missing_branches,
         )
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.data.close()
+
     def missing_formatted(self, branches=False):
         """The missing line numbers, formatted nicely.