def test_pick_agents_full(self): agents = {} for i in range(10): agents[i] = { 'id': i, 'mode': 'alone', 'node': 'uno', } picked = [set(a['id'] for a in arr) for arr in server._pick_agents(agents, None)] self.assertEqual([set(range(10))], picked)
def test_pick_agents_full(self): agents = {} for i in range(10): agents[i] = { 'id': i, 'mode': 'alone', 'node': 'uno', } picked = [set(a['id'] for a in arr) for arr in server._pick_agents(agents, None)] self.assertEqual([set(range(10))], picked)
def test_pick_agents_linear(self): agents = {} for i in range(10): agents[i] = { 'id': i, 'mode': 'alone', 'node': 'uno', } picked = [set(a['id'] for a in arr) for arr in server._pick_agents(agents, 'linear')] self.assertEqual([set(range(i + 1)) for i in range(0, 10)], picked)
def test_pick_agents_linear(self): agents = {} for i in range(10): agents[i] = { 'id': i, 'mode': 'alone', 'node': 'uno', } picked = [set(a['id'] for a in arr) for arr in server._pick_agents(agents, 'linear')] self.assertEqual([set(range(i + 1)) for i in range(0, 10)], picked)
def test_pick_agents_quadratic(self): agents = {} for i in range(10): agents[i] = { 'id': i, 'mode': 'alone', 'node': 'uno', } picked = [set(a['id'] for a in arr) for arr in server._pick_agents(agents, 'quadratic')] self.assertEqual([set(range(1)), set(range(2)), set(range(5)), set(range(10))], picked)
def test_pick_agents_quadratic(self): agents = {} for i in range(10): agents[i] = { 'id': i, 'mode': 'alone', 'node': 'uno', } picked = [set(a['id'] for a in arr) for arr in server._pick_agents(agents, 'quadratic')] self.assertEqual([set(range(1)), set(range(2)), set(range(5)), set(range(10))], picked)
def test_pick_agents_full_filter_slaves(self): agents = {} for i in range(10): agents['master_%s' % i] = { 'id': 'master_%s' % i, 'mode': 'master', 'node': 'uno', } agents['slave_%s' % i] = { 'id': 'slave_%s' % i, 'mode': 'slave', 'node': 'uno', } picked = [set(a['id'] for a in arr) for arr in server._pick_agents(agents, None)] self.assertEqual([set('master_%s' % i for i in range(10))], picked)
def test_pick_agents_full_filter_slaves(self): agents = {} for i in range(10): agents['master_%s' % i] = { 'id': 'master_%s' % i, 'mode': 'master', 'node': 'uno', } agents['slave_%s' % i] = { 'id': 'slave_%s' % i, 'mode': 'slave', 'node': 'uno', } picked = [set(a['id'] for a in arr) for arr in server._pick_agents(agents, None)] self.assertEqual([set('master_%s' % i for i in range(10))], picked)
def test_pick_agents_full_filter_minions(self): agents = {} for i in range(10): agents['primary_%s' % i] = { 'id': 'primary_%s' % i, 'mode': 'primary', 'node': 'uno', } agents['minion_%s' % i] = { 'id': 'minion_%s' % i, 'mode': 'minion', 'node': 'uno', } picked = [set(a['id'] for a in arr) for arr in server._pick_agents(agents, None)] self.assertEqual([set('primary_%s' % i for i in range(10))], picked)