Added concept 'isa' other_concept functionality

2019-12-21 22:02:07 +01:00
parent 2474b08150
commit a683d4cd42
13 changed files with 489 additions and 61 deletions
@@ -3,7 +3,7 @@ import os
 from os import path
 import shutil

-from core.builtin_concepts import BuiltinConcepts, ReturnValueConcept, UserInputConcept
+from core.builtin_concepts import BuiltinConcepts, ReturnValueConcept, UserInputConcept, ConceptAlreadyInSet
 from core.concept import Concept, PROPERTIES_TO_SERIALIZE, Property
 from core.sheerka import Sheerka, ExecutionContext
 from evaluators.MutipleSameSuccessEvaluator import MultipleSameSuccessEvaluator
@@ -600,3 +600,71 @@ def test_builtin_error_concept_are_errors():
    # only test a random one, it will be the same for the others
    sheerka = get_sheerka()
    assert not sheerka.is_success(sheerka.new(BuiltinConcepts.TOO_MANY_SUCCESS))
+
+
+def test_i_can_add_concept_to_set():
+    sheerka = get_sheerka(False, False)
+
+    foo = Concept("foo")
+    sheerka.set_id_if_needed(foo, False)
+
+    all_foos = Concept("all_foos")
+    sheerka.set_id_if_needed(all_foos, False)
+
+    context = get_context(sheerka)
+    res = sheerka.add_concept_to_set(context, foo, all_foos)
+
+    assert res.status
+    assert sheerka.isinstance(res.body, BuiltinConcepts.SUCCESS)
+
+    all_entries = get_sheerka(False, False).sdp.get("All_" + all_foos.id, None, False)
+    assert len(all_entries) == 1
+    assert foo.id in all_entries
+
+
+def test_i_can_add_several_concepts_to_set():
+    sheerka = get_sheerka(False, False)
+
+    foo1 = Concept("foo1")
+    sheerka.set_id_if_needed(foo1, False)
+
+    foo2 = Concept("foo1")
+    sheerka.set_id_if_needed(foo2, False)
+
+    all_foos = Concept("all_foos")
+    sheerka.set_id_if_needed(all_foos, False)
+
+    context = get_context(sheerka)
+    sheerka.add_concept_to_set(context, foo1, all_foos)
+    res = sheerka.add_concept_to_set(context, foo2, all_foos)
+
+    assert res.status
+    assert sheerka.isinstance(res.body, BuiltinConcepts.SUCCESS)
+
+    all_entries = get_sheerka(False, False).sdp.get("All_" + all_foos.id, None, False)
+    assert len(all_entries) == 2
+    assert foo1.id in all_entries
+    assert foo2.id in all_entries
+
+
+def test_i_cannot_add_the_same_concept_twice_in_a_set():
+    sheerka = get_sheerka()
+
+    foo = Concept("foo")
+    sheerka.set_id_if_needed(foo, False)
+
+    all_foos = Concept("all_foos")
+    sheerka.set_id_if_needed(all_foos, False)
+
+    context = get_context(sheerka)
+    sheerka.add_concept_to_set(context, foo, all_foos)
+    res = sheerka.add_concept_to_set(context, foo, all_foos)
+
+    assert not res.status
+    assert res.body == ConceptAlreadyInSet(foo, all_foos)
+
+    all_entries = sheerka.sdp.get("All_" + all_foos.id, None, False)
+    assert len(all_entries) == 1
+    assert foo.id in all_entries
+
+