def test_str_graph():
dsk = {"x": 1}
assert str_graph(dsk) == dsk
dsk = {("x", 1): (inc, 1)}
assert str_graph(dsk) == {str(("x", 1)): (inc, 1)}
dsk = {("x", 1): (inc, 1), ("x", 2): (inc, ("x", 1))}
assert str_graph(dsk) == {
str(("x", 1)): (inc, 1),
str(("x", 2)): (inc, str(("x", 1))),
}
dsks = [
{
"x": 1
},
{
("x", 1): (inc, 1),
("x", 2): (inc, ("x", 1))
},
{
("x", 1): (sum, [1, 2, 3]),
("x", 2): (sum, [("x", 1), ("x", 1)])
},
]
for dsk in dsks:
sdsk = str_graph(dsk)
keys = list(dsk)
skeys = [str(k) for k in keys]
assert all(isinstance(k, str) for k in sdsk)
assert dask.get(dsk, keys) == dask.get(sdsk, skeys)
def test_str_graph():
dsk = {b'x': 1}
assert str_graph(dsk) == dsk
dsk = {('x', 1): (inc, 1)}
assert str_graph(dsk) == {str(('x', 1)): (inc, 1)}
dsk = {('x', 1): (inc, 1), ('x', 2): (inc, ('x', 1))}
assert str_graph(dsk) == {
str(('x', 1)): (inc, 1),
str(('x', 2)): (inc, str(('x', 1)))
}
dsks = [{
'x': 1
}, {
('x', 1): (inc, 1),
('x', 2): (inc, ('x', 1))
}, {
('x', 1): (sum, [1, 2, 3]),
('x', 2): (sum, [('x', 1), ('x', 1)])
}]
for dsk in dsks:
sdsk = str_graph(dsk)
keys = list(dsk)
skeys = [str(k) for k in keys]
assert all(isinstance(k, (str, bytes)) for k in sdsk)
assert dask.get(dsk, keys) == dask.get(sdsk, skeys)
def test_str_graph():
dsk = {"x": 1}
assert str_graph(dsk) == dsk
dsk = {("x", 1): (inc, 1)}
assert str_graph(dsk) == {str(("x", 1)): (inc, 1)}
dsk = {("x", 1): (inc, 1), ("x", 2): (inc, ("x", 1))}
assert str_graph(dsk) == {
str(("x", 1)): (inc, 1),
str(("x", 2)): (inc, str(("x", 1))),
}
dsks = [
{"x": 1},
{("x", 1): (inc, 1), ("x", 2): (inc, ("x", 1))},
{("x", 1): (sum, [1, 2, 3]), ("x", 2): (sum, [("x", 1), ("x", 1)])},
]
for dsk in dsks:
sdsk = str_graph(dsk)
keys = list(dsk)
skeys = [str(k) for k in keys]
assert all(isinstance(k, str) for k in sdsk)
assert dask.get(dsk, keys) == dask.get(sdsk, skeys)
dsk = {("y", 1): (SubgraphCallable({"x": ("y", 1)}, "x", (("y", 1),)), (("z", 1),))}
dsk = str_graph(dsk, extra_values=(("z", 1),))
assert dsk["('y', 1)"][0].dsk["x"] == "('y', 1)"
assert dsk["('y', 1)"][1][0] == "('z', 1)"
def test_str_graph():
dsk = {'x': 1}
assert str_graph(dsk) == dsk
dsk = {('x', 1): (inc, 1)}
assert str_graph(dsk) == {str(('x', 1)): (inc, 1)}
dsk = {('x', 1): (inc, 1), ('x', 2): (inc, ('x', 1))}
assert str_graph(dsk) == {str(('x', 1)): (inc, 1),
str(('x', 2)): (inc, str(('x', 1)))}
dsks = [{'x': 1},
{('x', 1): (inc, 1), ('x', 2): (inc, ('x', 1))},
{('x', 1): (sum, [1, 2, 3]),
('x', 2): (sum, [('x', 1), ('x', 1)])}]
for dsk in dsks:
sdsk = str_graph(dsk)
keys = list(dsk)
skeys = [str(k) for k in keys]
assert all(isinstance(k, str) for k in sdsk)
assert dask.get(dsk, keys) == dask.get(sdsk, skeys)
