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_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"}