dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-24 06:15:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

fix: Resolve temporary file permission issues in test_models.py

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-11-19 14:48:06 -08:00
parent 2c81105bfc
commit a799851832
1 changed files with 13 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/basic/test_models.py
View file

@ -108,12 +108,14 @@ class TestModels(unittest.TestCase):
}, },
] ]
with tempfile.NamedTemporaryFile(mode="w", suffix=".yml") as tmp: # Write to a regular file instead of NamedTemporaryFile for better cross-platform compatibility
yaml.dump(test_settings, tmp) tmp = tempfile.mktemp(suffix=".yml")
tmp.flush() try:
with open(tmp, 'w') as f:
yaml.dump(test_settings, f)
# Register the test settings # Register the test settings
register_models([tmp.name]) register_models([tmp])
# Test that defaults are applied when no exact match # Test that defaults are applied when no exact match
model = Model("claude-3-5-sonnet-20240620") model = Model("claude-3-5-sonnet-20240620")
@ -131,6 +133,13 @@ class TestModels(unittest.TestCase):
model = Model("gpt-4") model = Model("gpt-4")
self.assertEqual(model.extra_params["extra_headers"]["Foo"], "bar") self.assertEqual(model.extra_params["extra_headers"]["Foo"], "bar")
self.assertEqual(model.extra_params["some_param"], "some value") self.assertEqual(model.extra_params["some_param"], "some value")
finally:
# Clean up the temporary file
import os
try:
os.unlink(tmp)
except OSError:
pass
if __name__ == "__main__": if __name__ == "__main__":