style: format test_main.py with black

This commit is contained in:
Paul Gauthier (aider) 2024-12-07 08:08:40 -08:00
parent e10d06cb47
commit af3d8dd6a7

View file

@ -681,11 +681,16 @@ class TestMain(TestCase):
def test_set_env_multiple(self): def test_set_env_multiple(self):
# Test setting multiple environment variables # Test setting multiple environment variables
with GitTemporaryDirectory(): with GitTemporaryDirectory():
main([ main(
"--set-env", "TEST_VAR1=value1", [
"--set-env", "TEST_VAR2=value2", "--set-env",
"--exit", "--yes" "TEST_VAR1=value1",
]) "--set-env",
"TEST_VAR2=value2",
"--exit",
"--yes",
]
)
self.assertEqual(os.environ.get("TEST_VAR1"), "value1") self.assertEqual(os.environ.get("TEST_VAR1"), "value1")
self.assertEqual(os.environ.get("TEST_VAR2"), "value2") self.assertEqual(os.environ.get("TEST_VAR2"), "value2")