test_action_serialization.py 822 B

12345678910111213141516
  1. import pytest
  2. from opendevin.action import action_from_dict, Action, AgentThinkAction
  3. def test_action_serialization_deserialization():
  4. original_action_dict = {
  5. 'action': 'think',
  6. 'args': {'thought': 'This is a test.'}
  7. }
  8. action_instance = action_from_dict(original_action_dict)
  9. assert isinstance(action_instance, Action), 'The action instance should be an instance of Action.'
  10. assert isinstance(action_instance, AgentThinkAction), 'The action instance should be an instance of AgentThinkAction.'
  11. serialized_action_dict = action_instance.to_dict()
  12. serialized_action_dict.pop('message')
  13. assert serialized_action_dict == original_action_dict, 'The serialized action should match the original action dict.'
  14. # Additional tests for various action subclasses can be included here