17 lines
650 B
Python
17 lines
650 B
Python
import unittest
|
|
|
|
from . import test_data
|
|
|
|
from sia import util
|
|
|
|
class UtilTest(unittest.TestCase):
|
|
def test_get_valid_root_elements_single(self):
|
|
valid_elements = util.get_valid_root_elements(test_data.echo_action_schema)
|
|
self.assertEqual(valid_elements, {'test_tag'})
|
|
|
|
def test_split_response_single_element(self):
|
|
response = "Some reasoning here\n<test_tag>content</test_tag>"
|
|
valid_elements = {'test_tag'}
|
|
result = util.split_response(response, valid_elements)
|
|
self.assertEqual(result.reasoning, "Some reasoning here")
|
|
self.assertEqual(result.actions, "<test_tag>content</test_tag>") |