diff --git a/harmonizer/ast_semantic_parser.py b/harmonizer/ast_semantic_parser.py index c35b403..66885f8 100644 --- a/harmonizer/ast_semantic_parser.py +++ b/harmonizer/ast_semantic_parser.py @@ -55,6 +55,7 @@ def __init__(self, vocabulary: Set[str]): "run": "power", "execute": "power", "raise": "power", + "save": "power", # JUSTICE (Order, Rules, Logic) "assert": "justice", "try": "justice", @@ -71,6 +72,8 @@ def __init__(self, vocabulary: Set[str]): "connect": "love", "merge": "love", "print": "love", # Communication is a form of Love + "user": "love", + "profile": "love", } self._node_map: Dict[ast.AST, str] = {} diff --git a/tests/test_parser.py b/tests/test_parser.py index 9d3ef1f..5ef17c7 100644 --- a/tests/test_parser.py +++ b/tests/test_parser.py @@ -70,22 +70,6 @@ def test_execution_contextual_override(parser): assert set(concepts) == expected_concepts -def test_execution_contextual_override(parser): - """ - Tests the contextual override for `_concepts_found.add`. - This should be mapped to 'wisdom', not 'community'. - """ - code = "self._concepts_found.add('new_concept')" - expected_concepts = {"wisdom"} - - import ast - - body = ast.parse(code).body - concepts = parser.get_execution_concepts(body) - - assert set(concepts) == expected_concepts - - def test_execution_method_call(parser): """Tests that a method call (e.g., db.query) is mapped correctly.""" code = "db.query('SELECT * FROM users')"