add risk tests

davidbailey · Feb 12, 2020 · 26c20d3 · 26c20d3
1 parent fe46c69
commit 26c20d3
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/tests/test_risk.py b/tests/test_risk.py
@@ -3,13 +3,24 @@
 """
 import unittest
 
-from rail import Control, Impact, Likelihood, Risk, Risks, ThreatEvent, ThreatSources, Tree, Vulnerability
+from rail import (
+    Control,
+    Impact,
+    Likelihood,
+    Risk,
+    Risks,
+    ThreatEvent,
+    ThreatSources,
+    Tree,
+    Vulnerability,
+)
 
 LAM = 0.5
 NAME = "Test Impact"
 MU = 100
 SIGMA = 10
 
+
 class TestRisks(unittest.TestCase):
     """
     Class to test a risk
@@ -22,7 +33,9 @@ def setUp(self):
         self.system = Tree(name="test tree")
         self.system.add_child("test child")
         self.controls = [Control("test control", 100, 0.1)]
-        self.vulnerability = Vulnerability(self.threat_event, self.system, self.controls)
+        self.vulnerability = Vulnerability(
+            self.threat_event, self.system, self.controls
+        )
         self.likelihood = Likelihood(LAM)
         self.impact = Impact(NAME, MU, SIGMA)
         self.risks = Risks()
@@ -32,7 +45,7 @@ def test_risks(self):
         """
         Test risks
         """
-        pass
+        self.assertEqual(self.risks.expected_loss_deterministic_mean(), 6.968547903331899e+63)
 
 
 if __name__ == "__main__":