Skip to content

Commit 453af7f

Browse files
HarshvMahawararmintaenzertng
authored andcommitted
Add fixture dict. for ai_package profile
Signed-off-by: HarshvMahawar <hv062727@gmail.com>
1 parent 6771cad commit 453af7f

1 file changed

Lines changed: 20 additions & 0 deletions

File tree

tests/spdx3/fixtures.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
SpdxDocument,
3434
Tool,
3535
)
36+
from spdx_tools.spdx3.model.ai.ai_package import AIPackage, SafetyRiskAssessmentType
3637
from spdx_tools.spdx3.model.dataset.dataset import ConfidentialityLevelType, Dataset, DatasetAvailabilityType
3738
from spdx_tools.spdx3.model.licensing import (
3839
CustomLicense,
@@ -316,6 +317,24 @@ def namespace_map_fixture(
316317
"relationship_type": RelationshipType.FIXED_IN,
317318
}
318319

320+
AIPACKAGE_DICT = {
321+
"energy_consumption": "energyConsumption",
322+
"standard_compliance": ["standardCompliance"],
323+
"limitation": "aIPackageLimitation",
324+
"type_of_model": ["typeOfModel"],
325+
"information_about_training": "informationAboutTraining",
326+
"information_about_application": "informationAboutApplication",
327+
"hyperparameter": {"aIPackageHypParaKey": "aIPackageHypParaValue"},
328+
"model_data_preprocessing": ["aImodelDataPreprocessing"],
329+
"model_explainability": ["aImodelExplainability"],
330+
"sensitive_personal_information": True,
331+
"metric_decision_threshold": {"metricDecisionThresholdKey": "metricDecisionThresholdValue"},
332+
"metric": {"aIMetricKey": "aIMetricValue"},
333+
"domain": ["aIDomain"],
334+
"autonomy_type": True,
335+
"safety_risk_assessment": SafetyRiskAssessmentType.LOW,
336+
}
337+
319338
ARTIFACT_DICT = {
320339
"originated_by": ["https://spdx.test/tools-python/originatedBy"],
321340
"supplied_by": ["https://spdx.test/tools-python/suppliedBy"],
@@ -443,6 +462,7 @@ def namespace_map_fixture(
443462
VEX_UNDER_INVESTIGATION_VULN_ASSESSMENT_RELATIONSHIP_DICT,
444463
],
445464
Vulnerability: [ELEMENT_DICT, VULNERABILITY_DICT],
465+
AIPackage: [AIPACKAGE_DICT, PACKAGE_DICT, ELEMENT_DICT, ARTIFACT_DICT, SOFTWARE_ARTIFACT_DICT],
446466
File: [ELEMENT_DICT, ARTIFACT_DICT, SOFTWARE_ARTIFACT_DICT, FILE_DICT],
447467
Package: [ELEMENT_DICT, ARTIFACT_DICT, SOFTWARE_ARTIFACT_DICT, PACKAGE_DICT],
448468
Snippet: [ELEMENT_DICT, ARTIFACT_DICT, SOFTWARE_ARTIFACT_DICT, SNIPPET_DICT],

0 commit comments

Comments
 (0)