Python _leanの例

コード例 #1

ファイル: test_optimize.py プロジェクト: kwmsmith/blaze

def test_add():
    expr = t.x + 1
    expr2, fields = _lean(expr, fields=set(["x"]))
    assert expr2.isidentical(expr)
    assert fields == {"x"}

    expr = (t.x + t.y).label("a")
    expr2, fields = _lean(expr, fields=set(["a"]))
    assert expr2.isidentical(expr)
    assert fields == {"x", "y"}

コード例 #2

def test_add():
    expr = t.x + 1
    expr2, fields = _lean(expr, fields=set(['x']))
    assert expr2.isidentical(expr)
    assert fields == set(['x'])

    expr = (t.x + t.y).label('a')
    expr2, fields = _lean(expr, fields=set(['a']))
    assert expr2.isidentical(expr)
    assert fields == set(['x', 'y'])

コード例 #3

ファイル: test_optimize.py プロジェクト: CaptainAL/Spyder

def test_add():
    expr = t.x + 1
    expr2, fields = _lean(expr, fields=set(['x']))
    assert expr2.isidentical(expr)
    assert fields == set(['x'])

    expr = (t.x + t.y).label('a')
    expr2, fields = _lean(expr, fields=set(['a']))
    assert expr2.isidentical(expr)
    assert fields == set(['x', 'y'])

コード例 #4

def test_lean_on_Symbol():
    assert _lean(t, fields=['x'])[0] == t[['x']]
    assert _lean(t, fields=['x', 'y', 'z', 'w'])[0] == t

コード例 #5

def test_summary():
    expr, fields = _lean(summary(a=t.x.sum(), b=t.y.sum()), fields=['a'])
    assert expr.isidentical(summary(a=t.x.sum()))

    assert fields == set(['x'])

コード例 #6

def test_relabel():
    expr = t.relabel(x='X', y='Y')
    expr2, fields = _lean(expr, fields=set(['X', 'z']))
    assert expr2.isidentical(t[['x', 'z']].relabel(x='X'))
    assert fields == set(['x', 'z'])

コード例 #7

def test_label():
    expr = t.x.label('foo')
    expr2, fields = _lean(expr, fields=set(['foo']))
    assert expr2.isidentical(expr)
    assert fields == set(['x'])

コード例 #8

ファイル: test_optimize.py プロジェクト: kwmsmith/blaze

def test_relabel():
    expr = t.relabel(x="X", y="Y")
    expr2, fields = _lean(expr, fields=set(["X", "z"]))
    assert expr2.isidentical(t[["x", "z"]].relabel(x="X"))
    assert fields == {"x", "z"}

コード例 #9

ファイル: test_optimize.py プロジェクト: kwmsmith/blaze

def test_lean_on_Symbol():
    assert _lean(t, fields=["x"])[0] == t[["x"]]
    assert _lean(t, fields=["x", "y", "z", "w"])[0] == t

コード例 #10

ファイル: test_optimize.py プロジェクト: kwmsmith/blaze

def test_label():
    expr = t.x.label("foo")
    expr2, fields = _lean(expr, fields=set(["foo"]))
    assert expr2.isidentical(expr)
    assert fields == {"x"}

コード例 #11

ファイル: test_optimize.py プロジェクト: kwmsmith/blaze

def test_summary():
    expr, fields = _lean(summary(a=t.x.sum(), b=t.y.sum()), fields=["a"])
    assert expr.isidentical(summary(a=t.x.sum()))
    assert fields == {"x"}

コード例 #12

ファイル: test_optimize.py プロジェクト: CaptainAL/Spyder

def test_relabel():
    expr = t.relabel(x='X', y='Y')
    expr2, fields = _lean(expr, fields=set(['X', 'z']))
    assert expr2.isidentical(t[['x', 'z']].relabel(x='X'))
    assert fields == set(['x', 'z'])

コード例 #13

ファイル: test_optimize.py プロジェクト: CaptainAL/Spyder

def test_label():
    expr = t.x.label('foo')
    expr2, fields =  _lean(expr, fields=set(['foo']))
    assert expr2.isidentical(expr)
    assert fields == set(['x'])