def test_figure_grid_solo_traces(self):
fig = figure_grid([[{
'data': [{
'type': 'table',
'header': {
'values': [['One', 'Two']]
}
}]
}, {
'data': [{
'type': 'parcoords',
'dimensions': [{
'values': [1, 2]
}]
}]
}]],
column_widths=[0.4, 0.6],
column_spacing=0)
# Validate resulting figure object
go.Figure(fig)
# Check domains
self.assertEqual(fig['data'][0]['type'], 'table')
self.assertEqual(fig['data'][0]['domain'], {
'x': [0, 0.4],
'y': [0, 1.0]
})
self.assertEqual(fig['data'][1]['type'], 'parcoords')
self.assertEqual(fig['data'][1]['domain'], {
'x': [0.4, 1.0],
'y': [0, 1.0]
})
def test_titles_converted_to_annotations(self):
fig = figure_grid([[{
'data': [{
'type': 'scatter',
'y': [1, 3, 2]
}],
'layout': {
'title': 'Scatter!'
}
}],
[{
'data': [{
'type': 'bar',
'y': [2, 3, 1]
}],
'layout': {
'title': 'Bar!'
}
}]])
# Validate resulting figure object
go.Figure(fig)
self.assertNotIn('title', fig['layout'])
self.assertEqual(len(fig['layout']['annotations']), 2)
self.assertEqual(fig['layout']['annotations'][0]['text'], 'Scatter!')
self.assertEqual(fig['layout']['annotations'][1]['text'], 'Bar!')
def test_titles_converted_to_annotations(self):
fig = figure_grid([[
{'data': [{'type': 'scatter',
'y': [1, 3, 2]}],
'layout': {
'width': 400,
'height': 400,
'title': 'Scatter!'
}}
], [
{'data': [{'type': 'bar',
'y': [2, 3, 1]}],
'layout': {
'width': 400,
'height': 400,
'title': 'Bar!'
}}
]], row_spacing=100)
# Validate resulting figure object
go.Figure(fig)
self.assertNotIn('title', fig['layout'])
self.assertEqual(len(fig['layout']['annotations']), 2)
self.assertEqual(fig['layout']['annotations'][0]['text'], 'Scatter!')
self.assertEqual(fig['layout']['annotations'][1]['text'], 'Bar!')
# Check width and height
self.assertEqual(fig['layout']['width'], 400)
self.assertEqual(fig['layout']['height'], 900)
def test_figure_grid_polar_subplots(self):
fig = figure_grid([[
{'data': [{'type': 'scatterpolar',
'theta': [0, 90], 'r': [0.5, 1.0]}]}
], [
{'data': [{'type': 'barpolar',
'theta': [90, 180], 'r': [1.0, 10.0]}],
'layout': {'polar': {'radialaxis': {'title': 'radial'}}}}
]], row_spacing=0.1)
# Validate resulting figure object
go.Figure(fig)
# Check domains
self.assertEqual(fig['data'][0]['type'], 'scatterpolar')
self.assertEqual(fig['data'][0]['subplot'], 'polar')
self.assertEqual(fig['layout']['polar']['domain'],
{'y': [0, 0.45], 'x': [0, 1.0]})
self.assertEqual(fig['data'][1]['type'], 'barpolar')
self.assertEqual(fig['data'][1]['subplot'], 'polar2')
self.assertEqual(fig['layout']['polar2']['domain'],
{'y': [0.55, 1.0], 'x': [0, 1.0]})
# Check that radial axis title stayed with the barpolar trace's polar
# subplot
self.assertEqual(fig['layout']['polar2']['radialaxis'],
{'title': 'radial'})
def test_figure_grid_solo_traces(self):
fig = figure_grid([[
{'data': [{'type': 'table',
'header': {'values': [['One', 'Two']]}}],
'layout': {
'width': 400,
'height': 1000,
}},
{'data': [{'type': 'parcoords',
'dimensions': [{'values': [1, 2]}]}],
'layout': {
'width': 600,
'height': 1000,
}}
]], row_spacing=0, column_spacing=0)
# Validate resulting figure object
go.Figure(fig)
# Check domains
self.assertEqual(fig['data'][0]['type'], 'table')
self.assertEqual(fig['data'][0]['domain'],
{'x': [0, 0.4], 'y': [0.0, 1.0]})
self.assertEqual(fig['data'][1]['type'], 'parcoords')
self.assertEqual(fig['data'][1]['domain'],
{'x': [0.4, 1.0], 'y': [0, 1.0]})
# Check width and height
self.assertEqual(fig['layout']['width'], 1000)
self.assertEqual(fig['layout']['height'], 1000)
def test_axis_matching_offset(self):
fig = figure_grid([[
{'data': [{'type': 'scatter',
'y': [1, 3, 2],
'xaxis': 'x',
'yaxis': 'y'},
{'type': 'bar',
'y': [2, 3, 1],
'xaxis': 'x2',
'yaxis': 'y2',
}],
'layout': {
'width': 400,
'height': 400,
'xaxis2': {'matches': 'x'},
'yaxis2': {'matches': 'y'},
}},
], [
{'data': [{'type': 'scatter',
'y': [1, 3, 2],
'xaxis': 'x',
'yaxis': 'y',
},
{'type': 'bar',
'y': [2, 3, 1],
'xaxis': 'x2',
'yaxis': 'y2',
}],
'layout': {
'width': 400,
'height': 400,
'xaxis2': {'matches': 'y'},
'yaxis2': {'matches': 'x'},
}},
]], column_spacing=0, row_spacing=0)
# Check axes that traces are associated with
self.assertEqual(fig['data'][0]['xaxis'], 'x')
self.assertEqual(fig['data'][0]['yaxis'], 'y')
self.assertEqual(fig['data'][1]['xaxis'], 'x2')
self.assertEqual(fig['data'][1]['yaxis'], 'y2')
self.assertEqual(fig['data'][2]['xaxis'], 'x3')
self.assertEqual(fig['data'][2]['yaxis'], 'y3')
self.assertEqual(fig['data'][3]['xaxis'], 'x4')
self.assertEqual(fig['data'][3]['yaxis'], 'y4')
# Check matches references
self.assertEqual(fig['layout']['xaxis'].get('matches', None), None)
self.assertEqual(fig['layout']['yaxis'].get('matches', None), None)
self.assertEqual(fig['layout']['xaxis2'].get('matches', None), 'x')
self.assertEqual(fig['layout']['yaxis2'].get('matches', None), 'y')
self.assertEqual(fig['layout']['xaxis3'].get('matches', None), None)
self.assertEqual(fig['layout']['yaxis3'].get('matches', None), None)
self.assertEqual(fig['layout']['xaxis4'].get('matches', None), 'y3')
self.assertEqual(fig['layout']['yaxis4'].get('matches', None), 'x3')
def test_width_height_with_spacing(self):
fig = figure_grid([[
{'data': [{'type': 'scatter',
'y': [1, 3, 2]}],
'layout': {
'width': 380,
'height': 380}}
,
{'data': [{'type': 'bar',
'y': [2, 3, 1]}],
'layout': {
'width': 380,
'height': 380}}
], [
{'data': [{'type': 'scatterpolar',
'theta': [0, 90], 'r': [0.5, 1.0]}],
'layout': {
'width': 380,
'height': 1140,
}},
{'data': [{'type': 'table',
'header': {'values': [['One', 'Two']]}}],
'layout': {
'width': 380,
'height': 1140}
}
]], column_spacing=40, row_spacing=80,
width=400, height=800)
# Check domains
expected_x_domains = [[0, 0.45], [0.55, 1]]
expected_y_domains = [[0, 0.225], [0.325, 1]]
# scatter
self.assertEqual(fig['layout']['xaxis']['domain'], expected_x_domains[0])
self.assertEqual(fig['layout']['yaxis']['domain'], expected_y_domains[0])
# bar
self.assertEqual(fig['layout']['xaxis2']['domain'], expected_x_domains[1])
self.assertEqual(fig['layout']['yaxis2']['domain'], expected_y_domains[0])
# scatterpolar
self.assertEqual(fig['layout']['polar']['domain']['x'], expected_x_domains[0])
self.assertEqual(fig['layout']['polar']['domain']['y'], expected_y_domains[1])
# table
self.assertEqual(fig['data'][3]['domain']['x'], expected_x_domains[1])
self.assertEqual(fig['data'][3]['domain']['y'], expected_y_domains[1])
# Check width and height
self.assertEqual(fig['layout']['width'], 400)
self.assertEqual(fig['layout']['height'], 800)
def test_images_stick_with_axis(self):
fig = figure_grid([[
{'data': [{'type': 'scatter',
'y': [1, 3, 2]}],
'layout': {
'width': 400,
'height': 400,
'images': [
{'source': 'One.png',
'xref': 'x', 'yref': 'y',
'x': 0, 'y': 0},
{'source': 'Two.png',
'xref': 'x', 'yref': 'y',
'x': 1, 'y': 0}
]}
}
,
{'data': [{'type': 'bar',
'y': [2, 3, 1]}],
'layout': {
'width': 400,
'height': 400,
'images': [
{'source': 'Three.png',
'xref': 'x', 'yref': 'y',
'x': 2, 'y': 0},
{'source': 'Four.png',
'xref': 'x', 'yref': 'y',
'x': 3, 'y': 0}
]}}
]])
# Validate resulting figure object
go.Figure(fig)
images = fig['layout']['images']
self.assertEqual(len(images), 4)
self.assertEqual(images[0]['source'], 'One.png')
self.assertEqual(images[0]['xref'], 'x')
self.assertEqual(images[0]['yref'], 'y')
self.assertEqual(images[1]['source'], 'Two.png')
self.assertEqual(images[1]['xref'], 'x')
self.assertEqual(images[1]['yref'], 'y')
self.assertEqual(images[2]['source'], 'Three.png')
self.assertEqual(images[2]['xref'], 'x2')
self.assertEqual(images[2]['yref'], 'y2')
self.assertEqual(images[3]['source'], 'Four.png')
self.assertEqual(images[3]['xref'], 'x2')
self.assertEqual(images[3]['yref'], 'y2')
def test_shapes_stick_with_axis(self):
fig = figure_grid([[
{'data': [{'type': 'scatter',
'y': [1, 3, 2]}],
'layout': {
'width': 400,
'height': 400,
'shapes': [
{'type': 'rect',
'xref': 'x', 'yref': 'y',
'x0': 0, 'y0': 0, 'x1': 1, 'y1': 1},
{'type': 'circle',
'xref': 'x', 'yref': 'y',
'x0': 1, 'y0': 1, 'x1': 2, 'y1': 2}
]}}
,
{'data': [{'type': 'bar',
'y': [2, 3, 1]}],
'layout': {
'width': 400,
'height': 400,
'shapes': [
{'type': 'line',
'xref': 'x', 'yref': 'y',
'x0': 2, 'y0': 0, 'x1': 1, 'y1': 3},
{'type': 'path',
'xref': 'x', 'yref': 'y',
'x0': 3, 'y0': 0, 'x1': 3, 'y1': 6}
]}}
]])
# Validate resulting figure object
go.Figure(fig)
shapes = fig['layout']['shapes']
self.assertEqual(len(shapes), 4)
self.assertEqual(shapes[0]['type'], 'rect')
self.assertEqual(shapes[0]['xref'], 'x')
self.assertEqual(shapes[0]['yref'], 'y')
self.assertEqual(shapes[1]['type'], 'circle')
self.assertEqual(shapes[1]['xref'], 'x')
self.assertEqual(shapes[1]['yref'], 'y')
self.assertEqual(shapes[2]['type'], 'line')
self.assertEqual(shapes[2]['xref'], 'x2')
self.assertEqual(shapes[2]['yref'], 'y2')
self.assertEqual(shapes[3]['type'], 'path')
self.assertEqual(shapes[3]['xref'], 'x2')
self.assertEqual(shapes[3]['yref'], 'y2')
def test_annotations_stick_with_axis(self):
fig = figure_grid([[
{'data': [{'type': 'scatter',
'y': [1, 3, 2]}],
'layout': {
'width': 400,
'height': 400,
'annotations': [
{'text': 'One',
'xref': 'x', 'yref': 'y',
'x': 0, 'y': 0},
{'text': 'Two',
'xref': 'x', 'yref': 'y',
'x': 1, 'y': 0}
]}}
,
{'data': [{'type': 'bar',
'y': [2, 3, 1]}],
'layout': {
'width': 400,
'height': 400,
'annotations': [
{'text': 'Three',
'xref': 'x', 'yref': 'y',
'x': 2, 'y': 0},
{'text': 'Four',
'xref': 'x', 'yref': 'y',
'x': 3, 'y': 0}
]}}
]])
# Validate resulting figure object
go.Figure(fig)
annotations = fig['layout']['annotations']
self.assertEqual(len(annotations), 4)
self.assertEqual(annotations[0]['text'], 'One')
self.assertEqual(annotations[0]['xref'], 'x')
self.assertEqual(annotations[0]['yref'], 'y')
self.assertEqual(annotations[1]['text'], 'Two')
self.assertEqual(annotations[1]['xref'], 'x')
self.assertEqual(annotations[1]['yref'], 'y')
self.assertEqual(annotations[2]['text'], 'Three')
self.assertEqual(annotations[2]['xref'], 'x2')
self.assertEqual(annotations[2]['yref'], 'y2')
self.assertEqual(annotations[3]['text'], 'Four')
self.assertEqual(annotations[3]['xref'], 'x2')
self.assertEqual(annotations[3]['yref'], 'y2')
def test_annotations_stick_with_axis(self):
fig = figure_grid([[
{'data': [{'type': 'scatter',
'y': [1, 3, 2]}],
'layout': {
'annotations': [
{'text': 'One',
'xref': 'x', 'yref': 'y',
'x': 0, 'y': 0},
{'text': 'Two',
'xref': 'x', 'yref': 'y',
'x': 1, 'y': 0}
]}}
,
{'data': [{'type': 'bar',
'y': [2, 3, 1]}],
'layout': {
'annotations': [
{'text': 'Three',
'xref': 'x', 'yref': 'y',
'x': 2, 'y': 0},
{'text': 'Four',
'xref': 'x', 'yref': 'y',
'x': 3, 'y': 0}
]}}
]])
# Validate resulting figure object
go.Figure(fig)
annotations = fig['layout']['annotations']
self.assertEqual(len(annotations), 4)
self.assertEqual(annotations[0]['text'], 'One')
self.assertEqual(annotations[0]['xref'], 'x')
self.assertEqual(annotations[0]['yref'], 'y')
self.assertEqual(annotations[1]['text'], 'Two')
self.assertEqual(annotations[1]['xref'], 'x')
self.assertEqual(annotations[1]['yref'], 'y')
self.assertEqual(annotations[2]['text'], 'Three')
self.assertEqual(annotations[2]['xref'], 'x2')
self.assertEqual(annotations[2]['yref'], 'y2')
self.assertEqual(annotations[3]['text'], 'Four')
self.assertEqual(annotations[3]['xref'], 'x2')
self.assertEqual(annotations[3]['yref'], 'y2')
def test_titles_converted_to_annotations(self):
fig = figure_grid([[
{'data': [{'type': 'scatter',
'y': [1, 3, 2]}],
'layout': {'title': 'Scatter!'}}
], [
{'data': [{'type': 'bar',
'y': [2, 3, 1]}],
'layout': {'title': 'Bar!'}}
]])
# Validate resulting figure object
go.Figure(fig)
self.assertNotIn('title', fig['layout'])
self.assertEqual(len(fig['layout']['annotations']), 2)
self.assertEqual(fig['layout']['annotations'][0]['text'], 'Scatter!')
self.assertEqual(fig['layout']['annotations'][1]['text'], 'Bar!')
def test_figure_grid_solo_traces(self):
fig = figure_grid([[
{'data': [{'type': 'table',
'header': {'values': [['One', 'Two']]}}]},
{'data': [{'type': 'parcoords',
'dimensions': [{'values': [1, 2]}]}]}
]], column_widths=[0.4, 0.6],
column_spacing=0)
# Validate resulting figure object
go.Figure(fig)
# Check domains
self.assertEqual(fig['data'][0]['type'], 'table')
self.assertEqual(fig['data'][0]['domain'],
{'x': [0, 0.4], 'y': [0, 1.0]})
self.assertEqual(fig['data'][1]['type'], 'parcoords')
self.assertEqual(fig['data'][1]['domain'],
{'x': [0.4, 1.0], 'y': [0, 1.0]})
