diff --git a/tests/basic/test_scripting.py b/tests/basic/test_scripting.py index 87e7c303e..a3df22405 100644 --- a/tests/basic/test_scripting.py +++ b/tests/basic/test_scripting.py @@ -27,12 +27,10 @@ class TestScriptingAPI(unittest.TestCase): # Assertions self.assertEqual(mock_send.call_count, 2) mock_send.assert_any_call( - [{"role": "user", "content": "make a script that prints hello world"}], - functions=None + [{"role": "user", "content": "make a script that prints hello world"}], functions=None ) mock_send.assert_any_call( - [{"role": "user", "content": "make it say goodbye"}], - functions=None + [{"role": "user", "content": "make it say goodbye"}], functions=None ) self.assertEqual(result1, "Changes applied successfully.") self.assertEqual(result2, "Changes applied successfully.") @@ -57,8 +55,7 @@ class TestScriptingAPI(unittest.TestCase): # Assertions mock_send.assert_called_once_with( - [{"role": "user", "content": "add a new function"}], - functions=None + [{"role": "user", "content": "add a new function"}], functions=None ) self.assertEqual(result, "New function added successfully.") self.assertTrue(io.yes) # Check that 'yes' is set to True