senaduka/old-riskletpy
1
0
Fork 0
Code Issues 5 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Promenjene vrednosti weight + likelihood

Browse Source
This commit is contained in:
Amir
2025-05-12 22:48:16 +02:00
parent 1109489ef3
commit cbd0832ca1
5 changed files with 69 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/core/tests/test_utils.py
View File

@@ -71,7 +71,7 @@ class UtilsTests(TestCase):
mock_openai.return_value = mock_client
mock_response = MagicMock()
control_lines = [f"{i} : 8 : 5" for i in range(1, 11)]
control_lines = [f"{i} : 4 : 3" for i in range(1, 11)]
mock_response.choices[0].message.content = "\n".join(control_lines)
mock_client.chat.completions.create.return_value = mock_response
@@ -100,7 +100,7 @@ class UtilsTests(TestCase):
self.assertEqual(calculate_aggregate_likelihood(controls), 6)
def test_map_weight_to_impact_likelihood(self):
impact, likelihood = map_weight_to_impact_likelihood(50, 30, 100)
impact, likelihood = map_weight_to_impact_likelihood(50, 30, 50)
self.assertAlmostEqual(impact, 5.0)
self.assertAlmostEqual(likelihood, 3.0)
@@ -123,7 +123,7 @@ class UtilsTests(TestCase):
def test_generate_risk_graph(self):
risks_with_controls = [
{'risk': {'id': 1}, 'impact': 5.0, 'likelihood': 3.0},
{'risk': {'id': 2}, 'impact': 7.0, 'likelihood': 4.0}
{'risk': {'id': 2}, 'impact': 4.0, 'likelihood': 4.0}
]
graph_data = generate_risk_graph(risks_with_controls)