add test

Author: kashif

tests/envs/test_textarena_environment.py

@@ -0,0 +1,28 @@
+import os
+import sys
+
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
+
+from envs.textarena_env.server.environment import TextArenaEnvironment
+from envs.textarena_env.models import TextArenaMessage
+
+
+def test_convert_messages_coalesces_consecutive_characters():
+    env = object.__new__(TextArenaEnvironment)
+
+    raw_messages = [
+        (0, "[", "PROMPT"),
+        (0, "GAME", "PROMPT"),
+        (0, "]", "PROMPT"),
+        (1, "A", "MESSAGE"),
+        (1, "B", "MESSAGE"),
+        (2, "!", "MESSAGE"),
+    ]
+
+    converted = env._convert_messages(raw_messages)
+
+    assert converted == [
+        TextArenaMessage(sender_id=0, content="[GAME]", category="PROMPT"),
+        TextArenaMessage(sender_id=1, content="AB", category="MESSAGE"),
+        TextArenaMessage(sender_id=2, content="!", category="MESSAGE"),
+    ]