Skip to content

Commit dfea707

Browse files
committed
fix tests
1 parent 4a7797e commit dfea707

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/unit/test_files_checks.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def test_check_jsonl_invalid_role(tmp_path: Path):
207207
report = check_file(file)
208208

209209
assert not report["is_check_passed"]
210-
assert "Found invalid role `invalid_role`" in report["message"]
210+
assert "Invalid role `invalid_role` in conversation" in report["message"]
211211

212212

213213
def test_check_jsonl_non_alternating_roles(tmp_path: Path):
@@ -257,7 +257,7 @@ def test_check_jsonl_missing_field_in_conversation(tmp_path: Path):
257257

258258
report = check_file(file)
259259
assert not report["is_check_passed"]
260-
assert "Field `content` is missing for a turn" in report["message"]
260+
assert "Missing required column `content`" in report["message"]
261261

262262

263263
def test_check_jsonl_wrong_turn_type(tmp_path: Path):
@@ -277,7 +277,7 @@ def test_check_jsonl_wrong_turn_type(tmp_path: Path):
277277
report = check_file(file)
278278
assert not report["is_check_passed"]
279279
assert (
280-
"Invalid format on line 1 of the input file. Expected a dictionary"
280+
"Invalid format on line 1 of the input file. The `messages` column must be a list of dicts."
281281
in report["message"]
282282
)
283283

0 commit comments

Comments
 (0)