Python optimizeの例

コード例 #1

ファイル: test_bag.py プロジェクト: PhanidharJammula/py

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

コード例 #2

ファイル: test_bag.py プロジェクト: xiaopge/dask

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

コード例 #3

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

コード例 #4

ファイル: test_bag.py プロジェクト: serazing/dask

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

コード例 #5

ファイル: test_bag.py プロジェクト: caseyclements/dask

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

コード例 #6

ファイル: test_bag.py プロジェクト: BabeNovelty/dask

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