def __init__(self, ground_weight=1.0, output=None):
self.graph = nx.Graph()
self.conjunctions = set()
self.ground_weight = ground_weight
self.nodes = OrderedSet()
self.edges = OrderedSet()
self._edge_matrix = None
self._conjunction_matrix = None
self._node_matrix = None
self._node_conjunctions = None
self._edge_conductance = None
self.output = output
def test_identity():
'''
Identity sets are just ranges of numbers.
'''
iset = IdentitySet(10)
eq_(iset[5], 5)
eq_(iset.index(2), 2)
eq_(len(iset), 10)
assert iset == OrderedSet(range(10))
iset = pickle.loads(pickle.dumps(iset))
eq_(iset[5], 5)
eq_(iset.index(2), 2)
eq_(len(iset), 10)
assert iset == OrderedSet(range(10))
def __init__(self, output=None):
self.nodes = OrderedSet()
self._node_matrix = None
self._node_conjunctions = None
self._fast_matrix_up = None
self._transition_matrix = None
self._fast_conjunctions = None
def load(node_file, matrix_up, matrix_down, conjunction_file):
trust = TrustNetwork()
nodes = divisi2.load(node_file)
trust.nodes = OrderedSet(nodes)
trust._fast_matrix_up = divisi2.load(matrix_up)
trust._transition_matrix = divisi2.load(matrix_down)
trust._fast_conjunctions = divisi2.load(conjunction_file)
return trust
def test_pickle():
'''
Test that OrderedSets can be pickled.
'''
s = OrderedSet(['dog', 'cat', 'banana'])
import cPickle as pickle
s2 = pickle.loads(pickle.dumps(s))
eq_(s, s2)
eq_(s2[0], 'dog')
eq_(s2.index('cat'), 1)
def test_delete_and_pickle():
'''
Deleting an element doesn't affect the remaining elements'
indices.
'''
s = OrderedSet(['dog', 'cat', 'banana'])
del s[1]
eq_(s[1], None)
eq_(s.index('banana'), 2)
# Pickling doesn't change things.
s2 = pickle.loads(pickle.dumps(s))
eq_(s, s2)
eq_(s2[1], None)
eq_(s2.index('banana'), 2)
assert None not in s2
assert None not in s2
def __init__(self, output=None):
self.nodes = OrderedSet()
self.node_objs = {} #dict mapping node name to Node object
self._node_conjunctions = None
self._fast_matrix = None
self._fast_conjunctions = None
def test_reprOfEmpty():
'''
repr() of an empty OrderedSet should not fail.
'''
repr(OrderedSet())
