def test_run3(self):
"""Doctest case with propage recovered singles"""
d = ContextAggregator(
config={
"propagation_mode": ContextAggregator.AGGREGATION_MODE,
"max_tau": 1,
"propagate_recovered_singles": True,
}
)
# d.initialize() # Always execute initialize before newly receive data
# Emulating receive data from neighbors
d.receive(1, set([Context(value=1.0, cohorts=[0, 1, 2])]))
d.receive(2, set([Context(value=2.0, cohorts=[0])]))
d.receive(3, set([Context(value=3.0, cohorts=[1])]))
d.receive(4, set([Context(value=7.0, cohorts=[9], hopcount=Context.SPECIAL_CONTEXT)]))
# Emulating accumulated contexts
context_db = set(
[
Context(value=1.0, cohorts=[2, 4, 5, 3]),
Context(value=1.0, cohorts=[5, 6]),
Context(value=7.0, cohorts=[7, 8]),
]
)
d.set_database(singles=set([]), aggregates=context_db, timestamp=10)
d.run_dataflow(timestamp=10)
# Emulating newly found singles and aggregates from database
self.assertTrue(same(d.get_database_singles(timestamp=10), [[0, 1, 2, 9], []]))
self.assertTrue(same(d.get_database_aggregates(timestamp=10), [[7, 8], [3, 4, 5], [6, 5]]))
# Emulating the disaggregation process
self.assertTrue(same(d.get_singles(timestamp=10), [[0, 1, 2, 9], []]))
self.assertTrue(same(d.get_primes(timestamp=10), [[7, 8]]))
self.assertTrue(same(d.get_non_primes(timestamp=10), [[3, 4, 5], [5, 6]]))
self.assertTrue(same(d.get_selected_non_primes(timestamp=10), [[3, 4, 5]]))
self.assertTrue(d.get_new_aggregate().get_cohorts_as_set() == set([0, 1, 2, 3, 4, 5, 7, 8, 9]))
self.assertTrue(same(d.get_filtered_singles(), [[0, 1, 2, 9], []]))
def test_run1(self):
"""Only [4,5] is prime
All the others are singles
"""
d = ContextAggregator()
d.receive(1, set([Context(value=1.0, cohorts=[0, 1, 2])]))
d.receive(2, set([Context(value=2.0, cohorts=[0])]))
d.set_database(
set([Context(value=1.0, cohorts=[0, 1, 2, 3, 4, 5]), Context(value=2.0, cohorts=[1])]),
set([Context(value=1.0, cohorts=[2, 3])]),
)
d.run_dataflow()
self.assertTrue(same(d.get_singles(), [[0, 1, 2, 3], []]))
self.assertTrue(same(d.get_primes(), [[], [4, 5]]))
