def test_prefer_broker_nodes():
"""
b0 b1 b2
| \ /
a0 a1
a1 should be run before a0
"""
a, b, c = 'abc'
dsk = {(a, 0): (f,), (a, 1): (f,),
(b, 0): (f, (a, 0)), (b, 1): (f, (a, 1)), (b, 2): (f, (a, 1))}
dependencies, dependents = get_deps(dsk)
nd = ndependents(dependencies, dependents)
cm = child_max(dependencies, dependents, nd)
o = order(dsk)
assert o[(a, 1)] < o[(a, 0)]
# Switch name of 0, 1 to ensure that this isn't due to string comparison
dsk = {(a, 0): (f,), (a, 1): (f,),
(b, 0): (f, (a, 0)), (b, 1): (f, (a, 1)), (b, 2): (f, (a, 0))}
o = order(dsk)
assert o[(a, 1)] > o[(a, 0)]
def test_deep_bases_win_over_dependents():
"""
d should come before e and probably before one of b and c
a
/ | \ .
b c |
/ \ | /
e d
"""
dsk = {
'a': (f, 'b', 'c', 'd'),
'b': (f, 'd', 'e'),
'c': (f, 'd'),
'd': 1,
'e': 2
}
dependencies, dependents = get_deps(dsk)
nd = ndependents(dependencies, dependents)
cm = child_max(dependencies, dependents, nd)
o = order(dsk)
assert o['d'] < o['e']
assert o['d'] < o['b'] or o['d'] < o['c']
def test_ndependents():
a, b, c = 'abc'
dsk = dict(chain((((a, i), i * 2) for i in range(5)),
(((b, i), (add, i, (a, i))) for i in range(5)),
(((c, i), (add, i, (b, i))) for i in range(5))))
result = ndependents(*get_deps(dsk))
expected = dict(chain((((a, i), 3) for i in range(5)),
(((b, i), 2) for i in range(5)),
(((c, i), 1) for i in range(5))))
assert result == expected
dsk = {a: 1, b: 1}
deps = get_deps(dsk)
assert ndependents(*deps) == dsk
dsk = {a: 1, b: (add, a, 1), c: (add, b, a)}
assert ndependents(*get_deps(dsk)) == {a: 4, b: 2, c: 1}
dsk = {a: 1, b: a, c: b}
deps = get_deps(dsk)
assert ndependents(*deps) == {a: 3, b: 2, c: 1}
def test_ndependents():
a, b, c = 'abc'
dsk = dict(chain((((a, i), i * 2) for i in range(5)),
(((b, i), (add, i, (a, i))) for i in range(5)),
(((c, i), (add, i, (b, i))) for i in range(5))))
result = ndependents(*get_deps(dsk))
expected = dict(chain((((a, i), 3) for i in range(5)),
(((b, i), 2) for i in range(5)),
(((c, i), 1) for i in range(5))))
assert result == expected
dsk = {a: 1, b: 1}
deps = get_deps(dsk)
assert ndependents(*deps) == dsk
dsk = {a: 1, b: (add, a, 1), c: (add, b, a)}
assert ndependents(*get_deps(dsk)) == {a: 4, b: 2, c: 1}
dsk = {a: 1, b: a, c: b}
deps = get_deps(dsk)
assert ndependents(*deps) == {a: 3, b: 2, c: 1}
def test_deep_bases_win_over_dependents():
"""
d should come before e and probably before one of b and c
a
/ | \ .
b c |
/ \ | /
e d
"""
dsk = {'a': (f, 'b', 'c', 'd'), 'b': (f, 'd', 'e'), 'c': (f, 'd'), 'd': 1,
'e': 2}
dependencies, dependents = get_deps(dsk)
nd = ndependents(dependencies, dependents)
cm = child_max(dependencies, dependents, nd)
o = order(dsk)
assert o['d'] < o['e']
assert o['d'] < o['b'] or o['d'] < o['c']
def test_prefer_broker_nodes():
"""
b0 b1 b2
| \ /
a0 a1
a1 should be run before a0
"""
a, b, c = 'abc'
dsk = {
(a, 0): (f, ),
(a, 1): (f, ),
(b, 0): (f, (a, 0)),
(b, 1): (f, (a, 1)),
(b, 2): (f, (a, 1))
}
dependencies, dependents = get_deps(dsk)
nd = ndependents(dependencies, dependents)
cm = child_max(dependencies, dependents, nd)
o = order(dsk)
assert o[(a, 1)] < o[(a, 0)]
# Switch name of 0, 1 to ensure that this isn't due to string comparison
dsk = {
(a, 0): (f, ),
(a, 1): (f, ),
(b, 0): (f, (a, 0)),
(b, 1): (f, (a, 1)),
(b, 2): (f, (a, 0))
}
o = order(dsk)
assert o[(a, 1)] > o[(a, 0)]
def test_stacklimit():
dsk = dict(('x%s' % (i + 1), (inc, 'x%s' % i)) for i in range(10000))
dependencies, dependents = get_deps(dsk)
scores = dict.fromkeys(dsk, 1)
child_max(dependencies, dependents, scores)
ndependents(dependencies, dependents)
def test_stacklimit():
dsk = dict(('x%s' % (i + 1), (inc, 'x%s' % i)) for i in range(10000))
dependencies, dependents = get_deps(dsk)
scores = dict.fromkeys(dsk, 1)
child_max(dependencies, dependents, scores)
ndependents(dependencies, dependents)
