def test_rename_fused_keys_bag(): inp = {"b": (list, "a"), "c": (f, "b", 1)} outp = optimize(inp, ["c"], rename_fused_keys=False) assert outp.keys() == {"c"} assert outp["c"][1:] == ("a", 1) with dask.config.set({"optimization.fuse.rename-keys": False}): assert optimize(inp, ["c"]) == outp # By default, fused keys are renamed assert optimize(inp, ["c"]) != outp
def test_inline_singleton_lists(): inp = {"b": (list, "a"), "c": (f, "b", 1)} out = {"c": (f, (list, "a"), 1)} assert inline_singleton_lists(inp, ["c"]) == out out = {"c": (f, "a", 1)} assert optimize(inp, ["c"], rename_fused_keys=False) == out # If list is an output key, don't fuse it assert inline_singleton_lists(inp, ["b", "c"]) == inp assert optimize(inp, ["b", "c"], rename_fused_keys=False) == inp inp = {"b": (list, "a"), "c": (f, "b", 1), "d": (f, "b", 2)} assert inline_singleton_lists(inp, ["c", "d"]) == inp # Doesn't inline constants inp = {"b": (4, 5), "c": (f, "b")} assert inline_singleton_lists(inp, ["c"]) == inp
def test_inline_singleton_lists(): inp = {'b': (list, 'a'), 'c': (f, 'b', 1)} out = {'c': (f, (list, 'a'), 1)} assert inline_singleton_lists(inp, ['c']) == out out = {'c': (f, 'a', 1)} assert optimize(inp, ['c'], rename_fused_keys=False) == out # If list is an output key, don't fuse it assert inline_singleton_lists(inp, ['b', 'c']) == inp assert optimize(inp, ['b', 'c'], rename_fused_keys=False) == inp inp = {'b': (list, 'a'), 'c': (f, 'b', 1), 'd': (f, 'b', 2)} assert inline_singleton_lists(inp, ['c', 'd']) == inp # Doesn't inline constants inp = {'b': (4, 5), 'c': (f, 'b')} assert inline_singleton_lists(inp, ['c']) == inp
def test_inline_singleton_lists(): inp = {'b': (list, 'a'), 'c': (f, 'b', 1)} out = {'c': (f, (list, 'a'), 1)} assert inline_singleton_lists(inp) == out out = {'c': (f, 'a', 1)} assert optimize(inp, ['c']) == out inp = {'b': (list, 'a'), 'c': (f, 'b', 1), 'd': (f, 'b', 2)} assert inline_singleton_lists(inp) == inp inp = {'b': (4, 5)} # doesn't inline constants assert inline_singleton_lists(inp) == inp
def test_inline_singleton_lists(): inp = {'b': (list, 'a'), 'c': (f, 'b', 1)} out = {'c': (f, (list, 'a'), 1)} assert inline_singleton_lists(inp) == out out = {'c': (f, 'a' , 1)} assert optimize(inp, ['c']) == out inp = {'b': (list, 'a'), 'c': (f, 'b', 1), 'd': (f, 'b', 2)} assert inline_singleton_lists(inp) == inp inp = {'b': (4, 5)} # doesn't inline constants assert inline_singleton_lists(inp) == inp