def testProb(self):
"""
probability attribute
"""
node = Event('Test')
with self.assertRaises(IndexError) as cm:
node.dist = ('LNORM', 'Fragility')
node.prob = ([3.78], 'Fragility', None, None, None, None)
self.assertIn('requires two inputs: Mean/Median and Standard_Deviaton',
str(cm.exception))
with self.assertRaises(IndexError) as cm:
node.dist = ('PE', 'risk')
node.prob = ([3.78, 1], 'risk', None, None, None, None)
self.assertIn('requires only one input: Mean/Median probability',
str(cm.exception))
with self.assertRaises(ValueError) as cm:
node.dist = ('PE', 'risk')
node.prob = ([3.78], 'risk', None, None, None, None)
self.assertIn('should be between 0 & 1', str(cm.exception))
with self.assertRaises(IndexError) as cm:
node.dist = ('NORM', 'risk')
node.prob = ([3.78], 'risk', None, None, None, None)
self.assertIn('requires two inputs: Mean/Median and Standard_Deviaton',
str(cm.exception))
def testString(self):
"""
test print operations
"""
node = Event('TOP', Event.OR)
node.children = [Event('B0'), Event('B1')]
string = str(node)
self.assertIn('TOP', string)
self.assertIn(' B0', string)
self.assertIn(' B1', string)
self.assertEqual(repr(node), 'TOP')
def testOperator(self):
"""
operator attribute
"""
node = Event('Test', Event.OR)
self.assertEqual(node.operator, Event.OR)
with self.assertRaises(TypeError) as cm:
node.operator = 42
self.assertIn('The operator must be', str(cm.exception))
node.operator = Event.AND
self.assertEqual(node.operator, Event.AND)
node.operator = None
self.assertEqual(node.operator, None)
def testBasicEvent(self):
"""
Basic node functions
"""
node = Event('Test')
self.assertEqual(node.name, 'Test')
self.assertEqual(node.operator, None)
with self.assertRaises(AttributeError) as cm:
node.name = 'NewName'
self.assertEqual("can't set attribute", str(cm.exception))
self.assertEqual(node, 'Test')
self.assertEqual(node, Event('Test'))
with self.assertRaises(TypeError) as cm:
node == 42 #pylint: disable=pointless-statement
self.assertEqual("Comparison with between 'int' type is not valid.",
str(cm.exception))
def testDist(self):
"""
distribution attribute
"""
node = Event('Test')
node.dist = ('LNORM', 'Fragility')
self.assertEqual(node.dist, 'LNORM')
node.dist = ('NORM', 'risk')
self.assertEqual(node.dist, 'NORM')
with self.assertRaises(TypeError) as cm:
node.dist = ('UNIF', 'Fragility')
self.assertIn('fragility must be one of the following',
str(cm.exception))
with self.assertRaises(TypeError) as cm:
node.dist = ('LNORM', 'risk')
self.assertIn('risk must be one of the following', str(cm.exception))
node = Event('Test', Event.OR)
with self.assertRaises(AttributeError) as cm:
node.dist = ('LNORM', 'Fragility')
self.assertIn("can not have distribution.", str(cm.exception))
def testChildren(self):
"""
children attribute
"""
node = Event('Test', Event.OR)
node.children = [Event('C0'), Event('C1')]
self.assertEqual(node.children[0], 'C0')
self.assertEqual(node.children[1], 'C1')
with self.assertRaises(TypeError) as cm:
node.children = 42 #pylint: disable=redefined-variable-type
self.assertEqual("The supplied items must be a list.",
str(cm.exception))
with self.assertRaises(TypeError) as cm:
node.children = [Event('C0'), 1980]
self.assertEqual("Supplied values must be Event objects.",
str(cm.exception))
with self.assertRaises(AttributeError) as cm:
node.operator = None
self.assertIn("The operator can not be set", str(cm.exception))
node = Event('Test')
with self.assertRaises(AttributeError) as cm:
node.children = [Event('C0'), Event('C1')]
self.assertIn("can not have children", str(cm.exception))
def assertMocus(self, mocus):
"""
Helper to make sure the mocus is correct.
"""
gold = [{Event('B2'), Event('B1')}, {Event('B4'),
Event('B1')},
{Event('B1'), Event('B5'),
Event('B3')}, {Event('B2'),
Event('B5'),
Event('B3')},
{Event('B4'), Event('B5'),
Event('B3')}]
self.assertEqual(len(mocus), len(gold))
for i in range(len(mocus)): #pylint: disable=consider-using-enumerate
self.assertEqual(sorted([n.name for n in mocus[i]]),
sorted([n.name for n in gold[i]]))