fix test

ngxson · ngxson · commit 101e772c73aa · 2024-12-17T22:28:53.000+01:00
diff --git a/examples/server/tests/unit/test_embedding.py b/examples/server/tests/unit/test_embedding.py
@@ -131,8 +131,8 @@ def test_same_prompt_give_same_result():
 @pytest.mark.parametrize(
     "content,n_tokens",
     [
-        ("I believe the meaning of life is", 7),
-        ("This is a test", 4),
+        ("I believe the meaning of life is", 9),
+        ("This is a test", 6),
     ]
 )
 def test_embedding_usage_single(content, n_tokens):
@@ -155,4 +155,4 @@ def test_embedding_usage_multiple():
     })
     assert res.status_code == 200
     assert res.body['usage']['prompt_tokens'] == res.body['usage']['total_tokens']
-    assert res.body['usage']['prompt_tokens'] == 2 * 7
+    assert res.body['usage']['prompt_tokens'] == 2 * 9

Original file line number	Diff line number	Diff line change
`@@ -131,8 +131,8 @@ def test_same_prompt_give_same_result():`
`131`	`131`	`@pytest.mark.parametrize(`
`132`	`132`	`"content,n_tokens",`
`133`	`133`	`[`
`134`		`- ("I believe the meaning of life is", 7),`
`135`		`- ("This is a test", 4),`
	`134`	`+ ("I believe the meaning of life is", 9),`
	`135`	`+ ("This is a test", 6),`
`136`	`136`	`]`
`137`	`137`	`)`
`138`	`138`	`def test_embedding_usage_single(content, n_tokens):`
`@@ -155,4 +155,4 @@ def test_embedding_usage_multiple():`
`155`	`155`	`})`
`156`	`156`	`assert res.status_code == 200`
`157`	`157`	`assert res.body['usage']['prompt_tokens'] == res.body['usage']['total_tokens']`
`158`		`- assert res.body['usage']['prompt_tokens'] == 2 * 7`
	`158`	`+ assert res.body['usage']['prompt_tokens'] == 2 * 9`