From f6be208d77b1d1f81c5f0dd61780ab79b03e8b51 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Tue, 11 Jul 2023 15:17:00 -0700 Subject: [PATCH] updated tests --- tests/test_editblock.py | 5 +---- tests/test_wholefile.py | 4 +--- 2 files changed, 2 insertions(+), 7 deletions(-) diff --git a/tests/test_editblock.py b/tests/test_editblock.py index 82940e3ef..bd5dc6e18 100644 --- a/tests/test_editblock.py +++ b/tests/test_editblock.py @@ -248,9 +248,7 @@ These changes replace the `subprocess.run` patches with `subprocess.check_output files = [file1] # Initialize the Coder object with the mocked IO and mocked repo - coder = Coder.create( - models.GPT4, "diff", io=InputOutput(), openai_api_key="fake_key", fnames=files - ) + coder = Coder.create(models.GPT4, "diff", io=InputOutput(), fnames=files) def mock_send(*args, **kwargs): coder.partial_response_content = f""" @@ -290,7 +288,6 @@ new models.GPT4, "diff", io=InputOutput(dry_run=True), - openai_api_key="fake_key", fnames=files, dry_run=True, ) diff --git a/tests/test_wholefile.py b/tests/test_wholefile.py index d6ff75b47..8f9f89fc5 100644 --- a/tests/test_wholefile.py +++ b/tests/test_wholefile.py @@ -292,9 +292,7 @@ after b files = [file1] # Initialize the Coder object with the mocked IO and mocked repo - coder = Coder.create( - models.GPT4, "whole", io=InputOutput(), openai_api_key="fake_key", fnames=files - ) + coder = Coder.create(models.GPT4, "whole", io=InputOutput(), fnames=files) # no trailing newline so the response content below doesn't add ANOTHER newline new_content = "new\ntwo\nthree"