Merge pull request #126 from strictdoc-project/value_map

Fixed enumeration value map

Author: stanislaw

reqif/models/reqif_data_type.py

@@ -143,10 +143,10 @@ def __init__(  # pylint: disable=too-many-arguments
         self.long_name: Optional[str] = long_name
         self.multi_valued: Optional[bool] = multi_valued
         self.values: Optional[List[ReqIFEnumValue]] = values
-        self.values_map: Dict[str, str] = {}
+        self.values_map: Dict[str, ReqIFEnumValue] = {}
         if values is not None:
             for value in values:
-                self.values_map[value.identifier] = value.key
+                self.values_map[value.identifier] = value
         self.is_self_closed: bool = is_self_closed
 
     @staticmethod

tests/unit/reqif/parsers/test_data_type_parser.py

@@ -58,7 +58,10 @@ def test_02_enumeration_type():
     assert isinstance(data_type, ReqIFDataTypeDefinitionEnumeration)
 
     assert data_type.identifier == "NODE_TYPE"
-    assert data_type.values_map == {
-        "NODE_TYPE_SECTION": "1",
-        "NODE_TYPE_REQUIREMENT": "2",
-    }
+    assert len(data_type.values_map) == 2
+
+    value_1 = data_type.values_map["NODE_TYPE_SECTION"]
+    value_2 = data_type.values_map["NODE_TYPE_REQUIREMENT"]
+
+    assert value_1.key == "1"
+    assert value_2.key == "2"