Add "one", "any" and "exists"

pyecoreocl/compiler.py

@@ -314,6 +314,28 @@ def rule_for_all(emitter, ctx):
     emitter.inline(")")
 
 
+@call_rule
+def rule_exists(emitter, ctx):
+    emitter.inline(f"any(")
+    emitter.visit(ctx.argExp().oclExp())
+    variables = [arg.text for arg in ctx.argExp().varnames]
+    emitter.inline(f" for {', '.join(variables)} in ")
+    if len(variables) > 1:
+        emitter.inline("itertools.combinations_with_replacement(")
+        emitter.visit(ctx.expression)
+        emitter.inline(f", {len(variables)})")
+    else:
+        emitter.visit(ctx.expression)
+    emitter.inline(")")
+
+
+@call_rule
+def rule_one(emitter, ctx):
+    emitter.inline("(len(list(")
+    rule_select(emitter, ctx)
+    emitter.inline(")) == 1)")
+
+
 @call_rule
 def rule_select(emitter, ctx):
     variables = [arg.text for arg in ctx.argExp().varnames]
@@ -467,6 +489,13 @@ def rule_as_bag(emitter, ctx):
     emitter.inline(")")
 
 
+@call_rule
+def rule_any(emitter, ctx):
+    emitter.inline("next(iter(")
+    emitter.visit(ctx.expression)
+    emitter.inline("), None)")
+
+
 def default_collection_call(emitter, ctx):
     operation = ctx.attname.text
     emitter.visit(ctx.expression)

tests/test_collection_lib.py

@@ -47,4 +47,25 @@ def test__collect():
     assert !l->collect(e: int | e + 1)->asSequence()! == [4, 5, 6]
 
     l = !Sequence{Tuple{value=3}, Tuple{value=4}, Tuple{value='stuff'}}!
-    assert !l->collect(e | e.value)->asSequence()! == [3, 4, "stuff"]
+    assert !l->collect(e | e.value)->asSequence()! == [3, 4, "stuff"]
+
+
+def test__any():
+    l = [3, 4, 5]
+    assert !l->any()! == 3
+
+    assert !Bag{3, 4, 5}->collect(e | e + 1)->any()! == 4
+
+
+def test__exists():
+    l = [3, 4, 5]
+    assert !l->exists(e | e = 3)! is True
+
+    assert !Bag{3, 4, 5}->collect(e | e + 1)->exists(e | e = 3)! is False
+
+
+def test__one():
+    l = [3, 4, 5]
+    assert !l->one(e | e = 3)! is True
+
+    assert !Bag{3, 4, 5}->collect(e | e + 1)->one(e | e = 3)! is False

tests/test_object_ptypes_lib.py

@@ -0,0 +1,4 @@
+# coding: magic_ocl.strict
+
+def test__trigger():
+    assert !true implies false! is False