def test_get_weights(self):
c1 = connectors.Connector(delays=0.5, weights=0.5)
self.assertEqual(c1.get_weights(3).tolist(), [0.5,0.5,0.5])
c2 = connectors.Connector(delays=0.5, weights="foo")
self.assertRaises(ValueError, c2.get_weights, 3)
class A(object): pass
c3 = connectors.Connector(delays=0.5, weights=A())
self.assertRaises(Exception,c3.get_weights, 3)
rd = random.RandomDistribution('gamma', [0.5,0.5])
c4 = connectors.Connector(delays=0.5, weights=rd)
w = c4.get_weights(3)
self.assertEqual(len(w), 3)
self.assertNotEqual(w[0], w[1])
def test_get_delays(self):
c1 = connectors.Connector(delays=0.5, weights=0.5)
self.assertEqual(c1.get_delays(3).tolist(), [0.5,0.5,0.5])
c2 = connectors.Connector(weights=0.5, delays="foo")
self.assertRaises(ValueError, c2.get_delays, 3)
class A(object): pass
c3 = connectors.Connector(weights=0.5, delays=A())
self.assertRaises(Exception,c3.get_delays, 3)
rd = random.RandomDistribution('gamma', [0.5,0.5])
c4 = connectors.Connector(weights=0.5, delays=rd)
d = c4.get_delays(3)
self.assertEqual(len(d), 3)
self.assertNotEqual(d[0], d[1])
c5 = connectors.Connector(weights=0.5, delays=[1.0, 2.0, 3.0])
self.assertEqual(c5.get_delays(3).tolist(), [1.0, 2.0, 3.0])
def test_connect(self):
c = connectors.Connector(delays=0.5)
self.assertRaises(NotImplementedError, c.connect, 'foo')
