Update unit tests

test/test_data.py

@@ -1,32 +1,31 @@
-echo_system_prompt = """
-Your answer always consists of 4 parts:
-    - The original request
-    - <test_tag>
-    - The original request
-    - </test_tag>
-    You never provide an answer.
-    Only the entered text, the xml open tag, the same text again and the xml close tag.
-    Don't add whitespace or newlines.
-    Don't modify the text or change casing.
-    Be exact, this is for testing purposes.
-
-    example input:
-    hello
-    example output:
-    hello<test_tag>hello</test_tag>
-""".strip()
-
-echo_action_schema = """
-<?xml version="1.0" encoding="UTF-8"?>
-<xs:schema xmlns:xs="http://www.w3.org/2001/XMLSchema">
-<xs:element name="test_tag">
-    <xs:complexType>
-        <xs:simpleContent>
-            <xs:extension base="xs:string">
-                <xs:attribute name="id" type="xs:string" use="required"/>
-            </xs:extension>
-        </xs:simpleContent>
-    </xs:complexType>
-</xs:element>
-</xs:schema>
+echo_system_prompt = """
+Your answer always consists of 3 parts:
+    - <test_tag>
+    - The original request
+    - </test_tag>
+    You never provide an answer.
+    Only the xml open tag, the original text and the xml close tag.
+    Don't add whitespace or newlines.
+    Don't modify the text or change casing.
+    Be exact, this is for testing purposes.
+
+    example input:
+    hello
+    example output:
+    <test_tag>hello</test_tag>
+""".strip()
+
+echo_action_schema = """
+<?xml version="1.0" encoding="UTF-8"?>
+<xs:schema xmlns:xs="http://www.w3.org/2001/XMLSchema">
+<xs:element name="test_tag">
+    <xs:complexType>
+        <xs:simpleContent>
+            <xs:extension base="xs:string">
+                <xs:attribute name="id" type="xs:string" use="required"/>
+            </xs:extension>
+        </xs:simpleContent>
+    </xs:complexType>
+</xs:element>
+</xs:schema>
 """.strip()