Skip to content

Commit 342ad3f

Browse files
better verbose output in cost test
Signed-off-by: thiswillbeyourgithub <[email protected]>
1 parent 8f511dd commit 342ad3f

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/test_wdoc.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def test_summary_tim_urban():
143143
)
144144
out = inst.summary_task()
145145
assert "urban" in out["summary"].lower() or "procrastinat" in out["summary"].lower()
146-
assert out["doc_total_cost"] > 0
146+
assert out["doc_total_cost"] > 0, out
147147

148148
inst2 = wdoc(
149149
task="summarize",
@@ -156,8 +156,9 @@ def test_summary_tim_urban():
156156
out2 = inst2.summary_task()
157157
assert "monkey" in out2["summary"].lower()
158158
assert (
159-
out2["doc_total_cost"] == 0
160-
), "Normally we should be reusing the cache so cost should be 0"
159+
out2["doc_total_cost"] == 0,
160+
out2,
161+
), f"Normally we should be reusing the cache so cost should be 0 but is {out2['doc_total_cost']}"
161162

162163
inst3 = wdoc(
163164
task="summarize",
@@ -171,7 +172,7 @@ def test_summary_tim_urban():
171172
assert "monkey" in out3["summary"].lower()
172173
assert (
173174
out3["doc_total_cost"] > 0
174-
), "Normally we disabled the cache so cost should be higher than 0"
175+
), f"Normally we disabled the cache so cost should be higher than 0 but is {out3['doc_total_cost']}"
175176
os.environ["WDOC_DISABLE_EMBEDDINGS_CACHE"] = "true"
176177

177178

0 commit comments

Comments
 (0)