Use TemporaryFile as a context manager

EliahKagan · EliahKagan · commit e40fc2cd2677 · 2024-02-21T00:41:41.000-05:00
(This also uses assertRaises as a context manager, but that is
just for improved clarity.)
diff --git a/test/test_git.py b/test/test_git.py
@@ -260,13 +260,9 @@ def test_it_ignores_false_kwargs(self, git):
         self.assertTrue("pass_this_kwarg" not in git.call_args[1])
 
     def test_it_raises_proper_exception_with_output_stream(self):
-        tmp_file = TemporaryFile()
-        self.assertRaises(
-            GitCommandError,
-            self.git.checkout,
-            "non-existent-branch",
-            output_stream=tmp_file,
-        )
+        with TemporaryFile() as tmp_file:
+            with self.assertRaises(GitCommandError):
+                self.git.checkout("non-existent-branch", output_stream=tmp_file)
 
     def test_it_accepts_environment_variables(self):
         filename = fixture_path("ls_tree_empty")
