fixing tests

davidbailey · Dec 9, 2019 · 98ddfe1 · 98ddfe1
1 parent 72c2dfc
commit 98ddfe1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tests/test_likelihood.py b/tests/test_likelihood.py
@@ -16,7 +16,7 @@ def test_likelihood(self):
         """
         Test a likelihood
         """
-        self.assertEqual(self.likelihood['name'], '.5')
+        self.assertEqual(self.likelihood['name'], '0.5')
         self.assertEqual(self.likelihood['lam'], .5)
 
 

diff --git a/tests/test_vulnerability.py b/tests/test_vulnerability.py
@@ -11,6 +11,7 @@ class TestVulnerabilities(unittest.TestCase):
     """
     def setUp(self):
         self.vulnerabilities = Vulnerabilities()
+        self.threat_sources = ThreatSources()
         self.threat_sources.new('test')
         self.threat_event = ThreatEvent('test event', self.threat_sources['test'])
         self.system = Tree(name='test tree')