def test_convert_y_success_temporal(column):
chart = ChartMetadata(alt.Chart(df).mark_point().encode(alt.Y(column)))
mapping = _convert(chart)
assert list(mapping['y']) == list(mdates.date2num(df[column].values))
def test_convert_x_success(channel):
chart_spec = alt.Chart(df).encode(x=channel).mark_point()
chart = ChartMetadata(chart_spec)
mapping = _convert(chart)
assert list(mapping['x']) == list(df[channel].values)
def test_convert_y_success(channel):
chart_spec = ChartMetadata(alt.Chart(df).encode(y=channel).mark_point())
mapping = _convert(chart_spec)
assert list(mapping['y']) == list(df[channel].values)
def test_convert_stroke_fail_temporal(column):
chart = ChartMetadata(
alt.Chart(df).mark_point().encode(alt.Stroke(column)))
_convert(chart)
def test_invalid_encodings():
chart_spec = alt.Chart(df).encode(x2='quant').mark_point()
chart = ChartMetadata(chart_spec)
with pytest.raises(ValueError):
_convert(chart)
def test_convert_size_success_nominal():
with pytest.raises(NotImplementedError):
chart_spec = ChartMetadata(
alt.Chart(df).encode(size='nom').mark_point())
_convert(chart_spec)
def test_quantitative_stroke():
chart = ChartMetadata(
alt.Chart(df_quant).mark_point().encode(alt.Stroke('fill')))
_convert(chart)
def test_convert_opacity_fail_temporal(column):
chart = ChartMetadata(
alt.Chart(df).mark_point().encode(alt.Opacity(column)))
_convert(chart)
def test_convert_size_success(channel, type):
chart_spec = ChartMetadata(
alt.Chart(df).encode(size='{}:{}'.format(channel, type)).mark_point())
mapping = _convert(chart_spec)
assert list(mapping['s']) == list(df[channel].values)
def test_quantitative_opacity_array():
chart = ChartMetadata(
alt.Chart(df_quant).mark_point().encode(alt.Opacity('alpha')))
_convert(chart)
def test_quantitative_shape():
chart = ChartMetadata(
alt.Chart(df_quant).mark_point().encode(alt.Shape('shape')))
mapping = _convert(chart)
def test_convert_color_success(channel, dtype):
chart_spec = ChartMetadata(
alt.Chart(df).encode(
color=alt.Color(field=channel, type=dtype)).mark_point())
mapping = _convert(chart_spec)
assert list(mapping['c']) == list(df[channel].values)
def test_convert_x2_y2_fail_temporal(column):
chart = ChartMetadata(
alt.Chart(df).mark_point().encode(alt.X2(column), alt.Y2(column)))
_convert(chart)
def test_quantitative_x2_y2():
chart = ChartMetadata(
alt.Chart(df_quant).mark_point().encode(alt.X('a'), alt.Y('b'),
alt.X2('c'), alt.Y2('alpha')))
_convert(chart)
