def test_path3D_multi_colors(self):
path3D = Path3D([[(0, 1, 0, 'red'), (1, 2, 1, 'red'),
(2, 3, 2, 'red')],
[(-1, 1, 3, 'blue'), (-2, 2, 4, 'blue'),
(-3, 3, 5, 'blue')]],
vdims='color').options(color='color')
state = self._get_plot_state(path3D)
self.assertEqual(state['data'][0]['line']['color'], 'red')
self.assertEqual(state['data'][1]['line']['color'], 'blue')
def test_path3D_state(self):
path3D = Path3D([(0, 1, 0), (1, 2, 1), (2, 3, 2)])
state = self._get_plot_state(path3D)
self.assertEqual(state['data'][0]['x'], np.array([0, 1, 2]))
self.assertEqual(state['data'][0]['y'], np.array([1, 2, 3]))
self.assertEqual(state['data'][0]['mode'], 'lines')
self.assertEqual(state['data'][0]['type'], 'scatter3d')
self.assertEqual(state['layout']['scene']['xaxis']['range'], [0, 2])
self.assertEqual(state['layout']['scene']['yaxis']['range'], [1, 3])
self.assertEqual(state['layout']['scene']['zaxis']['range'], [0, 2])
def test_path3D_multi(self):
path3D = Path3D([[(0, 1, 0), (1, 2, 1), (2, 3, 2)],
[(-1, 1, 3), (-2, 2, 4), (-3, 3, 5)]])
state = self._get_plot_state(path3D)
self.assertEqual(state['data'][0]['x'], np.array([0, 1, 2]))
self.assertEqual(state['data'][0]['y'], np.array([1, 2, 3]))
self.assertEqual(state['data'][0]['z'], np.array([0, 1, 2]))
self.assertEqual(state['data'][0]['mode'], 'lines')
self.assertEqual(state['data'][0]['type'], 'scatter3d')
self.assertEqual(state['data'][1]['x'], np.array([-1, -2, -3]))
self.assertEqual(state['data'][1]['y'], np.array([1, 2, 3]))
self.assertEqual(state['data'][1]['z'], np.array([3, 4, 5]))
self.assertEqual(state['data'][1]['mode'], 'lines')
self.assertEqual(state['data'][1]['type'], 'scatter3d')
self.assertEqual(state['layout']['scene']['xaxis']['range'], [-3, 2])
self.assertEqual(state['layout']['scene']['yaxis']['range'], [1, 3])
self.assertEqual(state['layout']['scene']['zaxis']['range'], [0, 5])
def test_overlay_plot_zlabel(self):
scatter = Path3D([]) * Scatter3D([(10, 1, 2), (100, 2, 3), (1000, 3, 5)
]).options(zlabel='Z-Axis')
state = self._get_plot_state(scatter)
self.assertEqual(state['layout']['scene']['zaxis']['title'], 'Z-Axis')
def test_overlay_plot_logz(self):
scatter = (Scatter3D([(0, 1, 10), (1, 2, 100),
(2, 3, 1000)]) * Path3D([])).options(logz=True)
state = self._get_plot_state(scatter)
self.assertEqual(state['layout']['scene']['zaxis']['type'], 'log')
def test_visible(self):
element = Path3D([(0, 1, 0), (1, 2, 1),
(2, 3, 2)]).options(visible=False)
state = self._get_plot_state(element)
self.assertEqual(state['data'][0]['visible'], False)