Hopiu/llm
1
0
Fork 0
mirror of https://github.com/Hopiu/llm.git synced 2026-04-23 14:34:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixed flaky order tests, refs #271

Browse source
This commit is contained in:
Simon Willison 2023-09-12 11:37:36 -07:00
parent e6dac1a1bd
commit 603da35e37
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/test_embed_cli.py
View file

@ -362,7 +362,7 @@ def test_embed_multi_sql(tmpdir, use_other_db, prefix):
assert result.exit_code == 0
embeddings_db = sqlite_utils.Database(db_path)
assert embeddings_db["embeddings"].count == 2
rows = list(embeddings_db.query("select id, content from embeddings"))
rows = list(embeddings_db.query("select id, content from embeddings order by id"))
assert rows == [
{"id": (prefix or "") + "1", "content": "cli Command line interface"},
{"id": (prefix or "") + "2", "content": "sql Structured query language"},
@ -432,20 +432,20 @@ def test_embed_multi_files(multi_files, scenario):
)
assert result.exit_code == 0
embeddings_db = sqlite_utils.Database(db_path)
rows = list(embeddings_db.query("select id, content from embeddings"))
rows = list(embeddings_db.query("select id, content from embeddings order by id"))
if scenario == "single":
assert rows == [
{"id": "file2.txt", "content": "goodbye world"},
{"id": "file1.txt", "content": "hello world"},
{"id": "nested/two.txt", "content": "two"},
{"id": "nested/one.txt", "content": "one"},
{"id": "file2.txt", "content": "goodbye world"},
{"id": "nested/more/three.txt", "content": "three"},
{"id": "nested/one.txt", "content": "one"},
{"id": "nested/two.txt", "content": "two"},
]
else:
assert rows == [
{"id": "ignored.ini", "content": "Has weird \x96 character"},
{"id": "two.txt", "content": "two"},
{"id": "one.txt", "content": "one"},
{"id": "two.txt", "content": "two"},
]
@ -487,7 +487,9 @@ def test_embed_multi_files_encoding(multi_files, extra_args, expected_error):
assert result.exit_code == 0
assert not result.stderr
embeddings_db = sqlite_utils.Database(db_path)
rows = list(embeddings_db.query("select id, content from embeddings"))
rows = list(
embeddings_db.query("select id, content from embeddings order by id")
)
assert rows == [
{"id": "ignored.ini", "content": "Has weird \x96 character"},
]