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]
# 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,
)

View file

@ -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"