def test_transforms():
# aggregate transform
chart = alt.Chart().transform_aggregate([], ['foo'])
kwds = {'aggregate': [], '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)]
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])])
def test_LookupData():
df = pd.DataFrame({'x': [1, 2, 3], 'y': [4, 5, 6]})
lookup = alt.LookupData(data=df, key='x')
dct = lookup.to_dict()
assert dct['key'] == 'x'
assert dct['data'] == {'values': [{'x': 1, 'y': 4},
{'x': 2, 'y': 5},
{'x': 3, 'y': 6}]}
def test_LookupData():
df = pd.DataFrame({'x': [1, 2, 3], 'y': [4, 5, 6]})
lookup = alt.LookupData(data=df, key='x')
with alt.data_transformers.enable(consolidate_datasets=False):
dct = lookup.to_dict()
assert dct['key'] == 'x'
assert dct['data'] == {
'values': [{
'x': 1,
'y': 4
}, {
'x': 2,
'y': 5
}, {
'x': 3,
'y': 6
}]
}
