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"}
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'])
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'])
def test_lean_on_Symbol():
assert _lean(t, fields=['x'])[0] == t[['x']]
assert _lean(t, fields=['x', 'y', 'z', 'w'])[0] == t
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'])
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'])
def test_label():
expr = t.x.label('foo')
expr2, fields = _lean(expr, fields=set(['foo']))
assert expr2.isidentical(expr)
assert fields == set(['x'])
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"}
def test_lean_on_Symbol():
assert _lean(t, fields=["x"])[0] == t[["x"]]
assert _lean(t, fields=["x", "y", "z", "w"])[0] == t
def test_label():
expr = t.x.label("foo")
expr2, fields = _lean(expr, fields=set(["foo"]))
assert expr2.isidentical(expr)
assert fields == {"x"}
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"}
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'])
def test_label():
expr = t.x.label('foo')
expr2, fields = _lean(expr, fields=set(['foo']))
assert expr2.isidentical(expr)
assert fields == set(['x'])
