code/self_healing_finetuning/tests/test_human_owner_options.py

from __future__ import annotations

import unittest

from orchestrator.human_owner_decision import _normalize_instruction


class HumanOwnerOptionTests(unittest.TestCase):
    def test_numeric_owner_options_map_to_actions(self) -> None:
        self.assertEqual(_normalize_instruction("1"), "continue")
        self.assertEqual(_normalize_instruction("[1]"), "continue")
        self.assertEqual(_normalize_instruction("2"), "exit")
        self.assertEqual(_normalize_instruction("[2]"), "exit")


if __name__ == "__main__":
    unittest.main()