updated tests

This commit is contained in:
Paul Gauthier 2023-07-11 15:17:00 -07:00
parent bcdf8968f1
commit f6be208d77
2 changed files with 2 additions and 7 deletions

View file

@ -248,9 +248,7 @@ These changes replace the `subprocess.run` patches with `subprocess.check_output
files = [file1] files = [file1]
# Initialize the Coder object with the mocked IO and mocked repo # Initialize the Coder object with the mocked IO and mocked repo
coder = Coder.create( coder = Coder.create(models.GPT4, "diff", io=InputOutput(), fnames=files)
models.GPT4, "diff", io=InputOutput(), openai_api_key="fake_key", fnames=files
)
def mock_send(*args, **kwargs): def mock_send(*args, **kwargs):
coder.partial_response_content = f""" coder.partial_response_content = f"""
@ -290,7 +288,6 @@ new
models.GPT4, models.GPT4,
"diff", "diff",
io=InputOutput(dry_run=True), io=InputOutput(dry_run=True),
openai_api_key="fake_key",
fnames=files, fnames=files,
dry_run=True, dry_run=True,
) )

View file

@ -292,9 +292,7 @@ after b
files = [file1] files = [file1]
# Initialize the Coder object with the mocked IO and mocked repo # Initialize the Coder object with the mocked IO and mocked repo
coder = Coder.create( coder = Coder.create(models.GPT4, "whole", io=InputOutput(), fnames=files)
models.GPT4, "whole", io=InputOutput(), openai_api_key="fake_key", fnames=files
)
# no trailing newline so the response content below doesn't add ANOTHER newline # no trailing newline so the response content below doesn't add ANOTHER newline
new_content = "new\ntwo\nthree" new_content = "new\ntwo\nthree"