Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly
Sign up
Appearance settings

Commit 36686ad

Browse files
authored
Merge pull request #460 from plotly/Gantt
First PR for Gantt Charts - Adding Tests Very Soon
2 parents 079c924 + a7be116 commit 36686ad

File tree

3 files changed

+1021
-0
lines changed

3 files changed

+1021
-0
lines changed

‎plotly/tests/test_core/test_tools/test_figure_factory.py‎

Lines changed: 240 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1125,6 +1125,246 @@ def test_table_with_index(self):
11251125
'zeroline': False}}}
11261126
self.assertEqual(index_table, exp_index_table)
11271127

1128+
1129+
class TestGantt(TestCase):
1130+
1131+
def test_validate_gantt(self):
1132+
1133+
# validate the basic gantt inputs
1134+
1135+
df = [{'Task': 'Job A',
1136+
'Start': '2009年02月01日',
1137+
'Finish': '2009年08月30日',
1138+
'Complete': 'a'}]
1139+
1140+
pattern2 = ('In order to use an indexing column and assign colors to '
1141+
'the values of the index, you must choose an actual '
1142+
'column name in the dataframe or key if a list of '
1143+
'dictionaries is being used.')
1144+
1145+
self.assertRaisesRegexp(PlotlyError, pattern2,
1146+
tls.FigureFactory.create_gantt,
1147+
df, index_col='foo')
1148+
1149+
df = 'foo'
1150+
1151+
pattern3 = ('You must input either a dataframe or a list of '
1152+
'dictionaries.')
1153+
1154+
self.assertRaisesRegexp(PlotlyError, pattern3,
1155+
tls.FigureFactory.create_gantt, df)
1156+
1157+
df = []
1158+
1159+
pattern4 = ('Your list is empty. It must contain at least one '
1160+
'dictionary.')
1161+
1162+
self.assertRaisesRegexp(PlotlyError, pattern4,
1163+
tls.FigureFactory.create_gantt, df)
1164+
1165+
df = ['foo']
1166+
1167+
pattern5 = ('Your list must only include dictionaries.')
1168+
1169+
self.assertRaisesRegexp(PlotlyError, pattern5,
1170+
tls.FigureFactory.create_gantt, df)
1171+
1172+
def test_gantt_index(self):
1173+
1174+
# validate the index used for gantt
1175+
1176+
df = [{'Task': 'Job A',
1177+
'Start': '2009年02月01日',
1178+
'Finish': '2009年08月30日',
1179+
'Complete': 50}]
1180+
1181+
pattern = ('In order to use an indexing column and assign colors to '
1182+
'the values of the index, you must choose an actual '
1183+
'column name in the dataframe or key if a list of '
1184+
'dictionaries is being used.')
1185+
1186+
self.assertRaisesRegexp(PlotlyError, pattern,
1187+
tls.FigureFactory.create_gantt,
1188+
df, index_col='foo')
1189+
1190+
df = [{'Task': 'Job A', 'Start': '2009年02月01日',
1191+
'Finish': '2009年08月30日', 'Complete': 'a'},
1192+
{'Task': 'Job A', 'Start': '2009年02月01日',
1193+
'Finish': '2009年08月30日', 'Complete': 50}]
1194+
1195+
pattern2 = ('Error in indexing column. Make sure all entries of each '
1196+
'column are all numbers or all strings.')
1197+
1198+
self.assertRaisesRegexp(PlotlyError, pattern2,
1199+
tls.FigureFactory.create_gantt,
1200+
df, index_col='Complete')
1201+
1202+
def test_gantt_validate_colors(self):
1203+
1204+
# validate the gantt colors variable
1205+
1206+
df = [{'Task': 'Job A', 'Start': '2009年02月01日',
1207+
'Finish': '2009年08月30日', 'Complete': 75, 'Resource': 'A'},
1208+
{'Task': 'Job B', 'Start': '2009年02月01日',
1209+
'Finish': '2009年08月30日', 'Complete': 50, 'Resource': 'B'}]
1210+
1211+
pattern = ('Whoops! The elements in your rgb colors tuples cannot '
1212+
'exceed 255.0.')
1213+
1214+
self.assertRaisesRegexp(PlotlyError, pattern,
1215+
tls.FigureFactory.create_gantt, df,
1216+
index_col='Complete', colors='rgb(300,1,1)')
1217+
1218+
self.assertRaises(PlotlyError, tls.FigureFactory.create_gantt,
1219+
df, index_col='Complete', colors='foo')
1220+
1221+
pattern2 = ('Whoops! The elements in your colors tuples cannot '
1222+
'exceed 1.0.')
1223+
1224+
self.assertRaisesRegexp(PlotlyError, pattern2,
1225+
tls.FigureFactory.create_gantt, df,
1226+
index_col='Complete', colors=(2, 1, 1))
1227+
1228+
# verify that if colors is a dictionary, its keys span all the
1229+
# values in the index column
1230+
colors_dict = {75: 'rgb(1, 2, 3)'}
1231+
1232+
pattern3 = ('If you are using colors as a dictionary, all of its '
1233+
'keys must be all the values in the index column.')
1234+
1235+
self.assertRaisesRegexp(PlotlyError, pattern3,
1236+
tls.FigureFactory.create_gantt, df,
1237+
index_col='Complete', colors=colors_dict)
1238+
1239+
# check: index is set if colors is a dictionary
1240+
colors_dict_good = {50: 'rgb(1, 2, 3)', 75: 'rgb(5, 10, 15)'}
1241+
1242+
pattern4 = ('Error. You have set colors to a dictionary but have not '
1243+
'picked an index. An index is required if you are '
1244+
'assigning colors to particular values in a dictioanry.')
1245+
1246+
self.assertRaisesRegexp(PlotlyError, pattern4,
1247+
tls.FigureFactory.create_gantt, df,
1248+
colors=colors_dict_good)
1249+
1250+
# check: number of colors is equal to or greater than number of
1251+
# unique index string values
1252+
pattern5 = ("Error. The number of colors in 'colors' must be no less "
1253+
"than the number of unique index values in your group "
1254+
"column.")
1255+
1256+
self.assertRaisesRegexp(PlotlyError, pattern5,
1257+
tls.FigureFactory.create_gantt, df,
1258+
index_col='Resource',
1259+
colors=['#ffffff'])
1260+
1261+
# check: if index is numeric, colors has at least 2 colors in it
1262+
pattern6 = ("You must use at least 2 colors in 'colors' if you "
1263+
"are using a colorscale. However only the first two "
1264+
"colors given will be used for the lower and upper "
1265+
"bounds on the colormap.")
1266+
1267+
self.assertRaisesRegexp(PlotlyError, pattern6,
1268+
tls.FigureFactory.create_gantt, df,
1269+
index_col='Complete',
1270+
colors=['#ffffff'])
1271+
1272+
def test_gantt_all_args(self):
1273+
1274+
# check if gantt chart matches with expected output
1275+
1276+
df = [{'Task': 'Run',
1277+
'Start': '2010年01月01日',
1278+
'Finish': '2011年02月02日',
1279+
'Complete': 0},
1280+
{'Task': 'Fast',
1281+
'Start': '2011年01月01日',
1282+
'Finish': '2012年06月05日',
1283+
'Complete': 25}]
1284+
1285+
test_gantt_chart = tls.FigureFactory.create_gantt(
1286+
df, colors='Blues', index_col='Complete', reverse_colors=True,
1287+
title='Title', bar_width=0.5, showgrid_x=True, showgrid_y=True,
1288+
height=500, width=500
1289+
)
1290+
1291+
exp_gantt_chart = {
1292+
'data': [{'marker': {'color': 'white'},
1293+
'name': '',
1294+
'x': ['2010年01月01日', '2011年02月02日'],
1295+
'y': [0, 0]},
1296+
{'marker': {'color': 'white'},
1297+
'name': '',
1298+
'x': ['2011年01月01日', '2012年06月05日'],
1299+
'y': [1, 1]}],
1300+
'layout': {'height': 500,
1301+
'hovermode': 'closest',
1302+
'shapes': [{'fillcolor': 'rgb(220.0, 220.0, 220.0)',
1303+
'line': {'width': 0},
1304+
'opacity': 1,
1305+
'type': 'rect',
1306+
'x0': '2010年01月01日',
1307+
'x1': '2011年02月02日',
1308+
'xref': 'x',
1309+
'y0': -0.5,
1310+
'y1': 0.5,
1311+
'yref': 'y'},
1312+
{'fillcolor': 'rgb(166.25, 167.5, 208.0)',
1313+
'line': {'width': 0},
1314+
'opacity': 1,
1315+
'type': 'rect',
1316+
'x0': '2011年01月01日',
1317+
'x1': '2012年06月05日',
1318+
'xref': 'x',
1319+
'y0': 0.5,
1320+
'y1': 1.5,
1321+
'yref': 'y'}],
1322+
'showlegend': False,
1323+
'title': 'Title',
1324+
'width': 500,
1325+
'xaxis': {'rangeselector': {'buttons': [
1326+
{'count': 7,
1327+
'label': '1w',
1328+
'step': 'day',
1329+
'stepmode': 'backward'},
1330+
{'count': 1,
1331+
'label': '1m',
1332+
'step': 'month',
1333+
'stepmode': 'backward'},
1334+
{'count': 6,
1335+
'label': '6m',
1336+
'step': 'month',
1337+
'stepmode': 'backward'},
1338+
{'count': 1,
1339+
'label': 'YTD',
1340+
'step': 'year',
1341+
'stepmode': 'todate'},
1342+
{'count': 1,
1343+
'label': '1y',
1344+
'step': 'year',
1345+
'stepmode': 'backward'},
1346+
{'step': 'all'}
1347+
]},
1348+
'showgrid': True,
1349+
'type': 'date',
1350+
'zeroline': False},
1351+
'yaxis': {'autorange': False,
1352+
'range': [-1, 3],
1353+
'showgrid': True,
1354+
'ticktext': ['Run', 'Fast'],
1355+
'tickvals': [0, 1],
1356+
'zeroline': False}}
1357+
}
1358+
1359+
self.assertEqual(test_gantt_chart['data'][0],
1360+
exp_gantt_chart['data'][0])
1361+
1362+
self.assertEqual(test_gantt_chart['data'][1],
1363+
exp_gantt_chart['data'][1])
1364+
1365+
self.assertEqual(test_gantt_chart['layout'],
1366+
exp_gantt_chart['layout'])
1367+
11281368
# class TestDistplot(TestCase):
11291369

11301370
# def test_scipy_import_error(self):

‎plotly/tests/test_optional/test_figure_factory.py‎

Lines changed: 79 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1267,6 +1267,85 @@ def test_scatter_plot_matrix_kwargs(self):
12671267
exp_scatter_plot_matrix['layout'])
12681268

12691269

1270+
class TestGantt(NumpyTestUtilsMixin, TestCase):
1271+
1272+
def test_df_dataframe(self):
1273+
1274+
# validate dataframe has correct column names
1275+
df1 = pd.DataFrame([[2, 'Apple']], columns=['Numbers', 'Fruit'])
1276+
self.assertRaises(PlotlyError, tls.FigureFactory.create_gantt, df1)
1277+
1278+
def test_df_dataframe_all_args(self):
1279+
1280+
# check if gantt chart matches with expected output
1281+
1282+
df = pd.DataFrame([['Job A', '2009年01月01日', '2009-02-30'],
1283+
['Job B', '2009年03月05日', '2009年04月15日']],
1284+
columns=['Task', 'Start', 'Finish'])
1285+
1286+
test_gantt_chart = tls.FigureFactory.create_gantt(df)
1287+
1288+
exp_gantt_chart = {
1289+
'data': [{'marker': {'color': 'white'},
1290+
'name': '',
1291+
'x': ['2009年01月01日', '2009-02-30'],
1292+
'y': [0, 0]}],
1293+
'layout': {'height': 600,
1294+
'hovermode': 'closest',
1295+
'shapes': [{'opacity': 1,
1296+
'y1': 0.2,
1297+
'xref': 'x',
1298+
'fillcolor': 'rgb(31.0, 119.0, 180.0)',
1299+
'yref': 'y',
1300+
'y0': -0.2,
1301+
'x0': '2009年01月01日',
1302+
'x1': '2009-02-30',
1303+
'type': 'rect',
1304+
'line': {'width': 0}},
1305+
{'opacity': 1,
1306+
'y1': 1.2,
1307+
'xref': 'x',
1308+
'fillcolor': 'rgb(255.0, 127.0, 14.0)',
1309+
'yref': 'y',
1310+
'y0': 0.8,
1311+
'x0': '2009年03月05日',
1312+
'x1': '2009年04月15日',
1313+
'type': 'rect',
1314+
'line': {'width': 0}}],
1315+
'showlegend': False,
1316+
'title': 'Gantt Chart',
1317+
'width': 900,
1318+
'xaxis': {'rangeselector': {'buttons': [
1319+
{'count': 7, 'label': '1w',
1320+
'step': 'day', 'stepmode': 'backward'},
1321+
{'count': 1, 'label': '1m',
1322+
'step': 'month', 'stepmode': 'backward'},
1323+
{'count': 6, 'label': '6m',
1324+
'step': 'month', 'stepmode': 'backward'},
1325+
{'count': 1, 'label': 'YTD',
1326+
'step': 'year', 'stepmode': 'todate'},
1327+
{'count': 1, 'label': '1y',
1328+
'step': 'year', 'stepmode': 'backward'},
1329+
{'step': 'all'}
1330+
]},
1331+
'showgrid': False,
1332+
'type': 'date',
1333+
'zeroline': False},
1334+
'yaxis': {'autorange': False,
1335+
'range': [-1, 3],
1336+
'showgrid': False,
1337+
'ticktext': ['Job A', 'Job B'],
1338+
'tickvals': [0, 1],
1339+
'zeroline': False}}
1340+
}
1341+
1342+
self.assertEqual(test_gantt_chart['data'][0],
1343+
exp_gantt_chart['data'][0])
1344+
1345+
self.assert_dict_equal(test_gantt_chart['layout'],
1346+
exp_gantt_chart['layout'])
1347+
1348+
12701349
class TestViolin(NumpyTestUtilsMixin, TestCase):
12711350

12721351
def test_colors_validation(self):

0 commit comments

Comments
(0)

AltStyle によって変換されたページ (->オリジナル) /