def test_convert_to_mpl_altair_datetime():
dates = [alt.DateTime(year=2015, date=7).to_dict(), alt.DateTime(year=2015, month="March", date=20).to_dict()]
assert list(_data._convert_to_mpl_date(dates)) == list(mdates.datestr2num(['2015-01-07', '2015-03-20']))
assert list(_data._convert_to_mpl_date(df_nonstandard['c'].values)) == list(mdates.datestr2num(df_nonstandard['c']))
def test_convert_to_mpl_datetime64():
assert list(_data._convert_to_mpl_date(df_nonstandard['e'].values)) == list(mdates.date2num(df_nonstandard['e']))
def test_convert_to_mpl_altair_datetime():
dates = [alt.DateTime(year=2015, date=7).to_dict(), alt.DateTime(year=2015, month="March", date=20).to_dict()]
assert list(_data._convert_to_mpl_date(dates)) == list(mdates.datestr2num(['2015-01-07', '2015-03-20']))
def test_convert_to_mpl_empty():
assert _data._convert_to_mpl_date([]) == []
@pytest.mark.parametrize('date,expected', [
(df_nonstandard['c'].values[0], mdates.datestr2num(df_nonstandard['c'].values[0])),
(df_nonstandard['e'].values[0], mdates.date2num(df_nonstandard['e'].values[0])),
(alt.DateTime(year=2015, month="March", date=7).to_dict(), mdates.datestr2num('2015-03-07'))
])
def test_convert_to_mpl_single_vals(date, expected):
assert _data._convert_to_mpl_date(date) == expected
@pytest.mark.parametrize('date,expected', [
(alt.DateTime(year=2015, month="March", date=7).to_dict(), '2015-03-07'),
(alt.DateTime(year=2015, date=7).to_dict(), '2015-01-07'),
(alt.DateTime(year=2015, month=3).to_dict(), '2015-03-01'),
(alt.DateTime(year=2015, date=7, milliseconds=1).to_dict(), '2015-01-07 00:00:00.001'),
pytest.param(alt.DateTime(day="Mon").to_dict(), '2015-01-07', marks=pytest.mark.xfail(raises=NotImplementedError)),
pytest.param(alt.DateTime(year=2015, date=20, quarter=1).to_dict(), '2015-01-20', marks=pytest.mark.xfail(raises=NotImplementedError)),
pytest.param(alt.DateTime(year=2015, date=20, utc=True).to_dict(), '2015-01-07', marks=pytest.mark.xfail(raises=NotImplementedError)),
pytest.param(alt.DateTime(date=20).to_dict(), '2015-01-07', marks=pytest.mark.xfail(raises=KeyError)),
])
def test_altair_datetime(date, expected):
del merge_gd["weeks_work"]
del merge_gd["num_fte"]
final = fact_table.merge(merge_gd,
right_on='task_id',
left_on='task_id',
how='left')
# Downsample
f1 = final["date"].dt.day % downsample_skip_days == 0
final = final[f1]
dead = pd.read_csv("deadlines.csv")
# Set up common x axis
dt1 = alt.DateTime(year=2017, month=12)
dt2 = alt.DateTime(year=2019, month=12)
x_scale = alt.Scale(domain=(dt1, dt2))
tt = [alt.Tooltip("num_fte", format=".2f"), alt.Tooltip("person")]
# dtt = alt.Tooltip("dead_desc")
# dtt2 = alt.Tooltip("person")
no_axis_title = axis = alt.Axis(title="")
alt_dead = alt.Chart(dead).mark_text(align="center",
baseline="middle",
size=20).encode(y=alt.Y('task_o:N'),
x=alt.X('start:T',
scale=x_scale),
text=alt.Text('mark'),
fig, ax = plt.subplots()
ax.scatter(**mapping)
convert_axis(ax, chart)
fig.tight_layout()
return fig
@pytest.mark.mpl_image_compare(baseline_dir='baseline_images/test_axis')
@pytest.mark.parametrize(
'df, x,y,x_dom,y_dom',
[(df_quant, 'a', 'c', [0.5, 4], [-5, 10]),
(df_quant, 'neg', 'alpha', [-6, -2], [0, 1]),
(df_temp, 'days', 'hrs', ['2014-12-25', '2015-03-01'
], ['2015-01-01', '2015-01-03']),
(df_temp, 'a', 'days', [0.5, 4], [
alt.DateTime(year=2014, month="Dec", date=25),
alt.DateTime(year=2015, month="March", date=1)
])])
def test_axis_domain(df, x, y, x_dom, y_dom):
chart = alt.Chart(df).mark_point().encode(
alt.X(x, scale=alt.Scale(domain=x_dom)),
alt.Y(y, scale=alt.Scale(domain=y_dom)))
mapping = convert(chart)
fig, ax = plt.subplots()
ax.scatter(**mapping)
convert_axis(ax, chart)
fig.tight_layout()
return fig
@pytest.mark.xfail(raises=NotImplementedError)
]
assert list(_data._convert_to_mpl_date(dates)) == list(
mdates.datestr2num(['2015-01-07', '2015-03-20']))
def test_convert_to_mpl_empty():
assert _data._convert_to_mpl_date([]) == []
@pytest.mark.parametrize(
'date,expected',
[(df_nonstandard['c'].values[0],
mdates.datestr2num(df_nonstandard['c'].values[0])),
(df_nonstandard['e'].values[0],
mdates.date2num(df_nonstandard['e'].values[0])),
(alt.DateTime(year=2015, month="March",
date=7).to_dict(), mdates.datestr2num('2015-03-07'))])
def test_convert_to_mpl_single_vals(date, expected):
assert _data._convert_to_mpl_date(date) == expected
@pytest.mark.parametrize('date,expected', [
(alt.DateTime(year=2015, month="March", date=7).to_dict(), '2015-03-07'),
(alt.DateTime(year=2015, date=7).to_dict(), '2015-01-07'),
(alt.DateTime(year=2015, month=3).to_dict(), '2015-03-01'),
(alt.DateTime(year=2015, date=7,
milliseconds=1).to_dict(), '2015-01-07 00:00:00.001'),
pytest.param(alt.DateTime(day="Mon").to_dict(),
'2015-01-07',
marks=pytest.mark.xfail(raises=NotImplementedError)),
pytest.param(alt.DateTime(year=2015, date=20, quarter=1).to_dict(),
'2015-01-20',
def to_altair_datetime(dt):
dt = pd.to_datetime(dt)
return alt.DateTime(year=dt.year, month=dt.month, date=dt.day)