new test structure. new project's methods for classes.

OntoUML · Dec 11, 2023 · 10ed85d · 10ed85d
1 parent aaafa53
commit 10ed85d
Show file tree

Hide file tree

Showing 20 changed files with 113 additions and 3 deletions.
diff --git a/ontouml_py/model/packageable.py b/ontouml_py/model/packageable.py
@@ -10,7 +10,6 @@
 from typing import Any
 from typing import Optional
 
-from icecream import ic
 from pydantic import PrivateAttr
 
 from ontouml_py.model.modelelement import ModelElement

diff --git a/ontouml_py/model/project.py b/ontouml_py/model/project.py
@@ -8,6 +8,7 @@
 from ontouml_py.model.anchor import Anchor
 from ontouml_py.model.binaryrelation import BinaryRelation
 from ontouml_py.model.class_py import Class
+from ontouml_py.model.enumerations.classstereotype import ClassStereotype
 from ontouml_py.model.enumerations.ontologyrepresentationstyle import OntologyRepresentationStyle
 from ontouml_py.model.generalization import Generalization
 from ontouml_py.model.generalizationset import GeneralizationSet
@@ -156,6 +157,8 @@ def get_shape_by_id(self, element_id: str):
     def get_view_by_id(self, element_id: str):
         return self.get_element_by_id("View", element_id)
 
+    ### ELEMENTS' CREATION CREATION METHODS
+
     def create_anchor(self, **data: dict[str, Any]):
         new_element = Anchor(self, **data)
         self._elements["Anchor"].add(new_element)
@@ -165,8 +168,8 @@ def create_binary_relation(self, **data: dict[str, Any]):
         self._elements["BinaryRelation"].add(new_element)
         return new_element
 
-    def create_class(self, **data: dict[str, Any]):
-        new_element = Class(self, **data)
+    def create_class(self, stereotype:Optional[ClassStereotype]=None, **data: dict[str, Any]):
+        new_element = Class(self, stereotype, **data)
         self._elements["Class"].add(new_element)
         return new_element
 
@@ -200,6 +203,114 @@ def create_package(self, **data: dict[str, Any]):
         self._elements["Package"].add(new_element)
         return new_element
 
+    ### CLASSES'S CREATION METHODS
+
+    def create_class_abstract(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.ABSTRACT
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_category(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.CATEGORY
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_collective(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.COLLECTIVE
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_datatype(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.DATATYPE
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_enumeration(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.ENUMERATION
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_event(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.EVENT
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_historical_role(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.HISTORICAL_ROLE
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_historical_role_mixin(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.HISTORICAL_ROLE_MIXIN
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_kind(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.KIND
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_mixin(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.MIXIN
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_mode(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.MODE
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_phase(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.PHASE
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_phase_mixin(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.PHASE_MIXIN
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_quality(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.QUALITY
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_quantity(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.QUANTITY
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_relator(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.RELATOR
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_role(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.ROLE
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_role_mixin(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.ROLE_MIXIN
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_situation(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.SITUATION
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_subkind(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.SUBKIND
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+    def create_class_type(self,**data: dict[str, Any]):
+        stereotype = ClassStereotype.TYPE
+        order = 1 # to be substituted
+        restricted_to = None # to be substituted
+        return self.create_class(stereotype=stereotype, order=order, restricted_to=restricted_to, **data)
+
     def delete_anchor(self, old_element: Anchor) -> None:
         if old_element not in self._elements["Anchor"]:
             raise ValueError(self.__deletion_error_message(old_element, "Anchor"))

diff --git a/tests/concrete_classes/__init__.py b/tests/concrete_classes/__init__.py
diff --git a/tests/concrete_classes/project/__init__.py b/tests/concrete_classes/project/__init__.py
diff --git a/tests/abstract_classes/__init__.py → tests/project/__init__.py b/tests/abstract_classes/__init__.py → tests/project/__init__.py
diff --git a/...lasses/project/test_project_attributes.py → tests/project/test_project_attributes.py b/...lasses/project/test_project_attributes.py → tests/project/test_project_attributes.py
diff --git a/...project/test_project_get_create_delete.py → ...project/test_project_get_create_delete.py b/...project/test_project_get_create_delete.py → ...project/test_project_get_create_delete.py
diff --git a/tests/abstract_classes/test_classifier.py → tests/test_classifier.py b/tests/abstract_classes/test_classifier.py → tests/test_classifier.py
diff --git a/tests/abstract_classes/test_decoratable.py → tests/test_decoratable.py b/tests/abstract_classes/test_decoratable.py → tests/test_decoratable.py
diff --git a/...s/concrete_classes/test_generalization.py → tests/test_generalization.py b/...s/concrete_classes/test_generalization.py → tests/test_generalization.py
diff --git a/...oncrete_classes/test_generalizationset.py → tests/test_generalizationset.py b/...oncrete_classes/test_generalizationset.py → tests/test_generalizationset.py
diff --git a/tests/concrete_classes/test_literal_class.py → tests/test_literal_class.py b/tests/concrete_classes/test_literal_class.py → tests/test_literal_class.py
diff --git a/tests/abstract_classes/test_modelelement.py → tests/test_modelelement.py b/tests/abstract_classes/test_modelelement.py → tests/test_modelelement.py
diff --git a/tests/abstract_classes/test_namedelement.py → tests/test_namedelement.py b/tests/abstract_classes/test_namedelement.py → tests/test_namedelement.py
diff --git a/tests/concrete_classes/test_note_anchor.py → tests/test_note_anchor.py b/tests/concrete_classes/test_note_anchor.py → tests/test_note_anchor.py
diff --git a/...s/abstract_classes/test_ontoumlelement.py → tests/test_ontoumlelement.py b/...s/abstract_classes/test_ontoumlelement.py → tests/test_ontoumlelement.py
diff --git a/...crete_classes/test_package_packageable.py → tests/test_package_packageable.py b/...crete_classes/test_package_packageable.py → tests/test_package_packageable.py
diff --git a/...te_classes/test_project_projectelement.py → tests/test_project_projectelement.py b/...te_classes/test_project_projectelement.py → tests/test_project_projectelement.py
diff --git a/tests/concrete_classes/test_property.py → tests/test_property.py b/tests/concrete_classes/test_property.py → tests/test_property.py
diff --git a/tests/abstract_classes/test_relation.py → tests/test_relation.py b/tests/abstract_classes/test_relation.py → tests/test_relation.py