def test_transforms():
# aggregate transform
agg1 = alt.AggregatedFieldDef(**{'as': 'x1', 'op': 'mean', 'field': 'y'})
agg2 = alt.AggregatedFieldDef(**{'as': 'x2', 'op': 'median', 'field': 'z'})
chart = alt.Chart().transform_aggregate([agg1], ['foo'], x2='median(z)')
kwds = dict(aggregate=[agg1, agg2], groupby=['foo'])
assert chart.transform == [alt.AggregateTransform(**kwds)]
# bin transform
chart = alt.Chart().transform_bin("binned", field="field", bin=True)
kwds = {'as': 'binned', 'field': 'field', 'bin': True}
assert chart.transform == [alt.BinTransform(**kwds)]
# calcualte transform
chart = alt.Chart().transform_calculate("calc", "datum.a * 4")
kwds = {'as': 'calc', 'calculate': 'datum.a * 4'}
assert chart.transform == [alt.CalculateTransform(**kwds)]
# filter transform
chart = alt.Chart().transform_filter("datum.a < 4")
assert chart.transform == [alt.FilterTransform(filter="datum.a < 4")]
# lookup transform
lookup_data = alt.LookupData(alt.UrlData('foo.csv'), 'id', ['rate'])
chart = alt.Chart().transform_lookup(from_=lookup_data,
as_='a',
lookup='a',
default='b')
kwds = {'from': lookup_data, 'as': 'a', 'lookup': 'a', 'default': 'b'}
assert chart.transform == [alt.LookupTransform(**kwds)]
# timeUnit transform
chart = alt.Chart().transform_timeunit("foo", field="x", timeUnit="date")
kwds = {'as': 'foo', 'field': 'x', 'timeUnit': 'date'}
assert chart.transform == [alt.TimeUnitTransform(**kwds)]
def test_transforms():
# aggregate transform
agg1 = alt.AggregatedFieldDef(**{'as': 'x1', 'op': 'mean', 'field': 'y'})
agg2 = alt.AggregatedFieldDef(**{'as': 'x2', 'op': 'median', 'field': 'z'})
chart = alt.Chart().transform_aggregate([agg1], ['foo'], x2='median(z)')
kwds = dict(aggregate=[agg1, agg2], groupby=['foo'])
assert chart.transform == [alt.AggregateTransform(**kwds)]
# bin transform
chart = alt.Chart().transform_bin("binned", field="field", bin=True)
kwds = {'as': 'binned', 'field': 'field', 'bin': True}
assert chart.transform == [alt.BinTransform(**kwds)]
# calcualte transform
chart = alt.Chart().transform_calculate("calc", "datum.a * 4")
kwds = {'as': 'calc', 'calculate': 'datum.a * 4'}
assert chart.transform == [alt.CalculateTransform(**kwds)]
# filter transform
chart = alt.Chart().transform_filter("datum.a < 4")
assert chart.transform == [alt.FilterTransform(filter="datum.a < 4")]
# lookup transform
lookup_data = alt.LookupData(alt.UrlData('foo.csv'), 'id', ['rate'])
chart = alt.Chart().transform_lookup(from_=lookup_data,
as_='a',
lookup='a',
default='b')
kwds = {'from': lookup_data, 'as': 'a', 'lookup': 'a', 'default': 'b'}
assert chart.transform == [alt.LookupTransform(**kwds)]
# timeUnit transform
chart = alt.Chart().transform_timeunit("foo", field="x", timeUnit="date")
kwds = {'as': 'foo', 'field': 'x', 'timeUnit': 'date'}
assert chart.transform == [alt.TimeUnitTransform(**kwds)]
# window transform
chart = alt.Chart().transform_window(xsum='sum(x)',
ymin='min(y)',
frame=[None, 0])
window = [
alt.WindowFieldDef(**{
'as': 'xsum',
'field': 'x',
'op': 'sum'
}),
alt.WindowFieldDef(**{
'as': 'ymin',
'field': 'y',
'op': 'min'
})
]
# kwargs don't maintain order in Python < 3.6, so window list can
# be reversed
assert (chart.transform
== [alt.WindowTransform(frame=[None, 0], window=window)]
or chart.transform
== [alt.WindowTransform(frame=[None, 0], window=window[::-1])])