v = contents[2].strip() link = frozenset([u, v]) time = (t, t) # This a new instance Cm.addClique(Clique((link, time), set([]))) # Populate data structures if link not in times: times[link] = [] times[link].append(t) if u not in nodes: nodes[u] = set() if v not in nodes: nodes[v] = set() nodes[u].add(v) nodes[v].add(u) nb_lines = nb_lines + 1 Cm._times = times Cm._nodes = nodes sys.stderr.write("Processed " + str(nb_lines) + " from stdin\n") # Restart execution R = Cm.getDeltaCliques(delta) sys.stdout.write("# delta = %d\n" % (delta)) Cm.printCliques()
def run(self): #graphTypes = ["Link stream", "Multiplex graph", "Simple graph"] self.delta = self.dataSet["Delta"] self.graph_type = self.dataSet["Input graph type"].getCurrentString() self.node_prop = self.dataSet["Node class - link stream"] self.one_boolean_property_per_clique = self.dataSet["Boolean property output"] #graphType = "Link stream" #if graph.getName() in graphTypes: # graphType = graph.getName() #node_prop = graph.getStringProperty("__original_node__") if self.graph_type == "Link stream" or self.graph_type == "Multiplex graph": self.time_stamp = self.dataSet["Time double property - Link stream / Multiplex graph"] #timeStamp = graph.getDoubleProperty("__timeStamp__") elif self.graph_type == "Simple graph": self.time_stamp = self.dataSet["Time vector property - Simple graph"] #timeStamp = graph.getDoubleVectorProperty("__timeStampList__") # Initiate Cm = CliqueMaster() times = dict() nodes = dict() nb_lines = 0 # Read stream # maybe just rewrite an individual accessor for 1 triplet # instead of the whole conversion # (needs the edge object for effective return) for contents in self.tripletGen(): #print contents t = contents[0] u = contents[1] v = contents[2] link = frozenset([u, v]) time = (t, t) Cm.addClique(Clique((link, time), set([]))) # Populate data structures if link not in times: times[link] = [] times[link].append(t) if u not in nodes: nodes[u] = set() if v not in nodes: nodes[v] = set() nodes[u].add(v) nodes[v].add(u) nb_lines = nb_lines + 1 Cm._times = times Cm._nodes = nodes tmp_std_err = sys.stderr sys.stderr = NullDevice() results = Cm.getDeltaCliques(self.delta) sys.stderr = tmp_std_err self.mapCliquesBack(results) return True
class TestClique(unittest.TestCase): def setUp(self): self.Cm = CliqueMaster() self.seq = range(10) sys.stderr = open(os.devnull, 'w') def test_delta_is_0(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))) ]) self.Cm._times = {frozenset([1, 2]): [1]} self.Cm._nodes = {1: set([2]), 2: set([1])} R = self.Cm.getDeltaCliques(0) self.assertEqual(R, set([Clique((frozenset([1, 2]), (1, 1)))])) def test_negative_delta(self): pass def test_big_delta(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))) ]) self.Cm._times = {frozenset([1, 2]): [1]} self.Cm._nodes = {1: set([2]), 2: set([1])} R = self.Cm.getDeltaCliques(100) self.assertEqual(R, set([Clique((frozenset([1, 2]), (-99, 101)))])) def test_simple_triangle_when_delta_is_5(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([1, 3]), (2, 2))), Clique((frozenset([2, 3]), (3, 3))) ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 3]): [2], frozenset([1, 2]): [1], frozenset([2, 3]): [3]} R = self.Cm.getDeltaCliques(5) R_expected = set([ Clique((frozenset([1, 2, 3]), (-2, 6))), Clique((frozenset([1, 2]), (-4, 6))), Clique((frozenset([2, 3]), (-2, 8))), Clique((frozenset([1, 3]), (-3, 7))) ]) debug_msg = "\nGot :\n" + str(self.Cm) debug_msg += "\nExpected :\n" for c in R_expected: debug_msg += str(c) + "\n" self.assertEqual(R, R_expected, debug_msg) def test_two_links_alterning(self): pass def test_single_link_occurring_every_delta(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([1, 2]), (3, 3))), ]) self.Cm._nodes = {1: set([2]), 2: set([1])} self.Cm._times = {frozenset([1, 2]): [1, 3]} R = self.Cm.getDeltaCliques(3) R_expected = set([ Clique((frozenset([1, 2]), (-2, 6))) ]) self.assertEqual(R, R_expected) def test_single_link_not_occurring_every_delta(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([1, 2]), (3, 3))), ]) self.Cm._nodes = {1: set([2]), 2: set([1])} self.Cm._times = {frozenset([1, 2]): [1, 3]} R = self.Cm.getDeltaCliques(1) R_expected = set([ Clique((frozenset([1, 2]), (0, 2))), Clique((frozenset([1, 2]), (2, 4))) ]) self.assertEqual(R, R_expected) def test_triangle_and_many_occurrences_with_delta_too_small(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([2, 3]), (2, 2))), Clique((frozenset([1, 3]), (3, 3))), Clique((frozenset([1, 2]), (4, 4))), ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 3]): [3], frozenset([1, 2]): [1, 4], frozenset([2, 3]): [2]} R = self.Cm.getDeltaCliques(2) R_expected = set([ Clique((frozenset([1, 2, 3]), (2, 4))), Clique((frozenset([1, 2]), (-1, 3))), Clique((frozenset([1, 3]), (1, 5))), Clique((frozenset([1, 2, 3]), (1, 3))), Clique((frozenset([1, 2]), (2, 6))), Clique((frozenset([2, 3]), (0, 4))) ]) debug_msg = "\nGot :\n" + str(self.Cm) debug_msg += "\nExpected :\n" for c in R_expected: debug_msg += str(c) + "\n" self.assertEqual(R, R_expected, debug_msg) def test_triangle_and_many_occurrences_with_delta_big(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([2, 3]), (2, 2))), Clique((frozenset([1, 3]), (3, 3))), Clique((frozenset([1, 2]), (4, 4))), ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 3]): [3], frozenset([1, 2]): [1, 4], frozenset([2, 3]): [2]} R = self.Cm.getDeltaCliques(5) R_expected = set([ Clique((frozenset([1, 2]), (-4, 9))), Clique((frozenset([1, 2, 3]), (-2, 7))), Clique((frozenset([1, 3]), (-2, 8))), Clique((frozenset([2, 3]), (-3, 7))) ]) self.assertEqual(R, R_expected) def test_triangle_and_many_occurrences_with_delta_huge(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([2, 3]), (2, 2))), Clique((frozenset([1, 3]), (3, 3))), Clique((frozenset([1, 2]), (4, 4))) ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 3]): [3], frozenset([1, 2]): [1, 4], frozenset([2, 3]): [2]} R = self.Cm.getDeltaCliques(100) R_expected = set([ Clique((frozenset([1, 2]), (-99, 104))), Clique((frozenset([1, 2, 3]), (-97, 102))), Clique((frozenset([1, 3]), (-97, 103))), Clique((frozenset([2, 3]), (-98, 102))) ]) debug_msg = "\nGot :\n" + str(self.Cm) debug_msg += "\nExpected :\n" for c in R_expected: debug_msg += str(c) + "\n" self.assertEqual(R, R_expected, debug_msg) def test_simultaneous_links(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([2, 3]), (1, 1))), Clique((frozenset([1, 3]), (1, 1))) ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 2]): [1], frozenset([2, 3]): [1], frozenset([1, 3]): [1]} R = self.Cm.getDeltaCliques(5) R_expected = set([Clique((frozenset([1, 2, 3]), (-4, 6)))]) self.assertEqual(R, R_expected) def test_simultaneous_links_with_repeat(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([2, 3]), (1, 1))), Clique((frozenset([1, 3]), (1, 1))), Clique((frozenset([2, 3]), (3, 3))) ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 2]): [1], frozenset([2, 3]): [1, 3], frozenset([1, 3]): [1]} R = self.Cm.getDeltaCliques(5) R_expected = set([ Clique((frozenset([1, 2, 3]), (-4, 6))), Clique((frozenset([2, 3]), (-4, 8))) ]) self.assertEqual(R, R_expected)
class TestClique(unittest.TestCase): def setUp(self): self.Cm = CliqueMaster() self.seq = range(10) sys.stderr = open(os.devnull, 'w') def test_delta_is_0(self): self.Cm._S = deque([Clique((frozenset([1, 2]), (1, 1)))]) self.Cm._times = {frozenset([1, 2]): [1]} self.Cm._nodes = {1: set([2]), 2: set([1])} R = self.Cm.getDeltaCliques(0) self.assertEqual(R, set([Clique((frozenset([1, 2]), (1, 1)))])) def test_negative_delta(self): pass def test_big_delta(self): self.Cm._S = deque([Clique((frozenset([1, 2]), (1, 1)))]) self.Cm._times = {frozenset([1, 2]): [1]} self.Cm._nodes = {1: set([2]), 2: set([1])} R = self.Cm.getDeltaCliques(100) self.assertEqual(R, set([Clique((frozenset([1, 2]), (-99, 101)))])) def test_simple_triangle_when_delta_is_5(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([1, 3]), (2, 2))), Clique((frozenset([2, 3]), (3, 3))) ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 3]): [2], frozenset([1, 2]): [1], frozenset([2, 3]): [3] } R = self.Cm.getDeltaCliques(5) R_expected = set([ Clique((frozenset([1, 2, 3]), (-2, 6))), Clique((frozenset([1, 2]), (-4, 6))), Clique((frozenset([2, 3]), (-2, 8))), Clique((frozenset([1, 3]), (-3, 7))) ]) debug_msg = "\nGot :\n" + str(self.Cm) debug_msg += "\nExpected :\n" for c in R_expected: debug_msg += str(c) + "\n" self.assertEqual(R, R_expected, debug_msg) def test_two_links_alterning(self): pass def test_single_link_occurring_every_delta(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([1, 2]), (3, 3))), ]) self.Cm._nodes = {1: set([2]), 2: set([1])} self.Cm._times = {frozenset([1, 2]): [1, 3]} R = self.Cm.getDeltaCliques(3) R_expected = set([Clique((frozenset([1, 2]), (-2, 6)))]) self.assertEqual(R, R_expected) def test_single_link_not_occurring_every_delta(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([1, 2]), (3, 3))), ]) self.Cm._nodes = {1: set([2]), 2: set([1])} self.Cm._times = {frozenset([1, 2]): [1, 3]} R = self.Cm.getDeltaCliques(1) R_expected = set([ Clique((frozenset([1, 2]), (0, 2))), Clique((frozenset([1, 2]), (2, 4))) ]) self.assertEqual(R, R_expected) def test_triangle_and_many_occurrences_with_delta_too_small(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([2, 3]), (2, 2))), Clique((frozenset([1, 3]), (3, 3))), Clique((frozenset([1, 2]), (4, 4))), ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 3]): [3], frozenset([1, 2]): [1, 4], frozenset([2, 3]): [2] } R = self.Cm.getDeltaCliques(2) R_expected = set([ Clique((frozenset([1, 2, 3]), (2, 4))), Clique((frozenset([1, 2]), (-1, 3))), Clique((frozenset([1, 3]), (1, 5))), Clique((frozenset([1, 2, 3]), (1, 3))), Clique((frozenset([1, 2]), (2, 6))), Clique((frozenset([2, 3]), (0, 4))) ]) debug_msg = "\nGot :\n" + str(self.Cm) debug_msg += "\nExpected :\n" for c in R_expected: debug_msg += str(c) + "\n" self.assertEqual(R, R_expected, debug_msg) def test_triangle_and_many_occurrences_with_delta_big(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([2, 3]), (2, 2))), Clique((frozenset([1, 3]), (3, 3))), Clique((frozenset([1, 2]), (4, 4))), ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 3]): [3], frozenset([1, 2]): [1, 4], frozenset([2, 3]): [2] } R = self.Cm.getDeltaCliques(5) R_expected = set([ Clique((frozenset([1, 2]), (-4, 9))), Clique((frozenset([1, 2, 3]), (-2, 7))), Clique((frozenset([1, 3]), (-2, 8))), Clique((frozenset([2, 3]), (-3, 7))) ]) self.assertEqual(R, R_expected) def test_triangle_and_many_occurrences_with_delta_huge(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([2, 3]), (2, 2))), Clique((frozenset([1, 3]), (3, 3))), Clique((frozenset([1, 2]), (4, 4))) ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 3]): [3], frozenset([1, 2]): [1, 4], frozenset([2, 3]): [2] } R = self.Cm.getDeltaCliques(100) R_expected = set([ Clique((frozenset([1, 2]), (-99, 104))), Clique((frozenset([1, 2, 3]), (-97, 102))), Clique((frozenset([1, 3]), (-97, 103))), Clique((frozenset([2, 3]), (-98, 102))) ]) debug_msg = "\nGot :\n" + str(self.Cm) debug_msg += "\nExpected :\n" for c in R_expected: debug_msg += str(c) + "\n" self.assertEqual(R, R_expected, debug_msg) def test_simultaneous_links(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([2, 3]), (1, 1))), Clique((frozenset([1, 3]), (1, 1))) ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 2]): [1], frozenset([2, 3]): [1], frozenset([1, 3]): [1] } R = self.Cm.getDeltaCliques(5) R_expected = set([Clique((frozenset([1, 2, 3]), (-4, 6)))]) self.assertEqual(R, R_expected) def test_simultaneous_links_with_repeat(self): self.Cm._S = deque([ Clique((frozenset([1, 2]), (1, 1))), Clique((frozenset([2, 3]), (1, 1))), Clique((frozenset([1, 3]), (1, 1))), Clique((frozenset([2, 3]), (3, 3))) ]) self.Cm._nodes = {1: set([2, 3]), 2: set([1, 3]), 3: set([1, 2])} self.Cm._times = { frozenset([1, 2]): [1], frozenset([2, 3]): [1, 3], frozenset([1, 3]): [1] } R = self.Cm.getDeltaCliques(5) R_expected = set([ Clique((frozenset([1, 2, 3]), (-4, 6))), Clique((frozenset([2, 3]), (-4, 8))) ]) self.assertEqual(R, R_expected)