add tests

Author: fdrose

tests/unit/test_files_checks.py

@@ -182,7 +182,7 @@ def test_check_jsonl_inconsistent_dataset_format(tmp_path: Path):
     # Create a JSONL file with inconsistent dataset formats
     file = tmp_path / "inconsistent_format.jsonl"
     content = [
-        {"messages": [{"role": "user", "content": "Hi"}]},
+        {"messages": [{"role": "user", "content": "Hi"}, {"role": "assistant", "content": "Hi! How can I help you?"}]},
         {"text": "How are you?"},  # Missing 'messages'
     ]
     with file.open("w") as f:
@@ -230,6 +230,19 @@ def test_check_jsonl_non_alternating_roles(tmp_path: Path):
     assert "Invalid role turns" in report["message"]
 
 
+def test_check_jsonl_non_alternating_roles(tmp_path: Path):
+    # Create a JSONL file with non-alternating user/assistant roles
+    file = tmp_path / "non_alternating_roles.jsonl"
+    content = [{"messages": [{"role": "user", "content": "Hi"}]}]
+    with file.open("w") as f:
+        f.write("\n".join(json.dumps(item) for item in content))
+
+    report = check_file(file)
+
+    assert not report["is_check_passed"]
+    assert "At least one message with the assistant role must be present" in report["message"]
+
+
 def test_check_jsonl_invalid_value_type(tmp_path: Path):
     # Create a JSONL file with an invalid value type
     file = tmp_path / "invalid_value_type.jsonl"