def test_frame_broadcast(): bcast = broadcast_collect(expr=t.amount * t.id) result = compute(bcast, df) assert_series_equal(result, df.amount * df.id)
def test_broadcast(): bcast = broadcast_collect(expr=t.amount * t.id) result = compute(bcast, ddf) eq(result, df.amount * df.id)
def test_series_broadcast(): s = Series([1, 2, 3], name='a') t = symbol('t', 'var * {a: int64}') bcast = broadcast_collect(expr=(2 * t.a - t.a**2)) result = compute(bcast, s) assert_series_equal(result, 2 * s - s**2)