Update tests

Author: connermanuel

tests/unit/test_files_checks.py

@@ -334,7 +334,7 @@ def test_check_jsonl_missing_field_in_conversation(tmp_path: Path):
         {
             "messages": [
                 {"role": "user", "content": "Hi"},
-                {"role": "assistant"},
+                {"content": "Hello"},
             ]
         }
     ]
@@ -343,7 +343,7 @@ def test_check_jsonl_missing_field_in_conversation(tmp_path: Path):
 
     report = check_file(file)
     assert not report["is_check_passed"]
-    assert "Missing required column `content`" in report["message"]
+    assert "Missing required column `role`" in report["message"]
 
 
 def test_check_jsonl_wrong_turn_type(tmp_path: Path):

tests/unit/test_preference_openai.py

@@ -149,12 +149,6 @@ def test_check_jsonl_invalid_preference_openai_missing_fields(tmp_path: Path, fi
         "Wrong format for non_preferred_output",
         id="wrong_output_format_non_preferred",
     ),
-    pytest.param(
-        "missing_content",
-        lambda item: item.update({"input": {"messages": [{"role": "user"}]}}),  # type: ignore
-        "Missing content in messages",
-        id="missing_content",
-    ),
     pytest.param(
         "multiple_preferred_outputs",
         lambda item: item.update(  # type: ignore