def get_jitter_plot():
boston = datasets.load_boston()
prices = boston['target']
x, y = get_data_sources(y=prices)
x_mapper, y_mapper = get_mappers(x, y)
jitter_plot = JitterPlot(
index=y,
mapper=y_mapper,
marker='circle',
jitter_width=100,
**PLOT_DEFAULTS
)
jitter_plot.line_width = 1.
x_axis = PlotAxis(orientation='bottom',
title='Median house prices',
mapper=jitter_plot.mapper,
component=jitter_plot,
**AXIS_DEFAULTS)
jitter_plot.underlays.append(x_axis)
return jitter_plot
def setUp(self):
self.size = (250, 250)
data_source = ArrayDataSource(arange(10))
index_range = DataRange1D()
index_range.add(data_source)
index_mapper = LinearMapper(range=index_range)
self.scatterplot = JitterPlot(
index=data_source,
index_mapper=index_mapper,
border_visible=False,
)
self.scatterplot.outer_bounds = list(self.size)
def setUp(self):
self.size = (250, 250)
data_source = ArrayDataSource(arange(10))
index_range = DataRange1D()
index_range.add(data_source)
index_mapper = LinearMapper(range=index_range)
self.scatterplot = JitterPlot(
index=data_source,
index_mapper=index_mapper,
border_visible=False,
)
self.scatterplot.outer_bounds = list(self.size)
class Jitterplot1DTest(unittest.TestCase):
def setUp(self):
self.size = (250, 250)
data_source = ArrayDataSource(arange(10))
index_range = DataRange1D()
index_range.add(data_source)
index_mapper = LinearMapper(range=index_range)
self.scatterplot = JitterPlot(
index=data_source,
index_mapper=index_mapper,
border_visible=False,
)
self.scatterplot.outer_bounds = list(self.size)
def test_scatter_1d(self):
self.assertEqual(self.scatterplot.origin, 'bottom left')
self.assertIsNone(self.scatterplot.x_mapper)
self.assertEqual(self.scatterplot.y_mapper,
self.scatterplot.index_mapper)
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_horizontal(self):
self.scatterplot.orientation = 'h'
self.assertEqual(self.scatterplot.origin, 'bottom left')
self.assertEqual(self.scatterplot.x_mapper,
self.scatterplot.index_mapper)
self.assertIsNone(self.scatterplot.y_mapper)
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_flipped(self):
self.scatterplot.direction = 'flipped'
self.assertEqual(self.scatterplot.origin, 'top left')
self.assertIsNone(self.scatterplot.x_mapper)
self.assertEqual(self.scatterplot.y_mapper,
self.scatterplot.index_mapper)
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_horizontal_flipped(self):
self.scatterplot.direction = 'flipped'
self.scatterplot.orientation = 'h'
self.assertEqual(self.scatterplot.origin, 'bottom right')
self.assertEqual(self.scatterplot.x_mapper,
self.scatterplot.index_mapper)
self.assertIsNone(self.scatterplot.y_mapper)
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_circle(self):
self.scatterplot.marker = 'circle'
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_custom(self):
path = CompiledPath()
path.move_to(-5, -5)
path.line_to(5, 5)
path.line_to(5, -5)
path.line_to(-5, 5)
path.line_to(-5, -5)
self.scatterplot.marker = 'custom'
self.scatterplot.custom_symbol = path
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_map_data(self):
points = array([[0, 124.5], [124.5, 0]])
assert_almost_equal(self.scatterplot.map_data(points),
array([4.5, 0]))
def test_scatter_1d_map_data_horizontal(self):
self.scatterplot.orientation = 'h'
points = array([[0, 124.5], [124.5, 0]])
assert_almost_equal(self.scatterplot.map_data(points),
array([0, 4.5]))
def test_scatter_1d_map_data_flipped(self):
self.scatterplot.direction = 'flipped'
points = array([[0, 124.5], [124.5, 0]])
assert_almost_equal(self.scatterplot.map_data(points),
array([4.5, 9.0]))
def test_scatter_1d_map_data_horizontal_flipped(self):
self.scatterplot.direction = 'flipped'
self.scatterplot.orientation = 'h'
points = array([[0, 124.5], [124.5, 0]])
assert_almost_equal(self.scatterplot.map_data(points),
array([9.0, 4.5]))
class Jitterplot1DTest(unittest.TestCase):
def setUp(self):
self.size = (250, 250)
data_source = ArrayDataSource(arange(10))
index_range = DataRange1D()
index_range.add(data_source)
index_mapper = LinearMapper(range=index_range)
self.scatterplot = JitterPlot(
index=data_source,
index_mapper=index_mapper,
border_visible=False,
)
self.scatterplot.outer_bounds = list(self.size)
def test_scatter_1d(self):
self.assertEqual(self.scatterplot.origin, 'bottom left')
self.assertIsNone(self.scatterplot.x_mapper)
self.assertEqual(self.scatterplot.y_mapper,
self.scatterplot.index_mapper)
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_horizontal(self):
self.scatterplot.orientation = 'h'
self.assertEqual(self.scatterplot.origin, 'bottom left')
self.assertEqual(self.scatterplot.x_mapper,
self.scatterplot.index_mapper)
self.assertIsNone(self.scatterplot.y_mapper)
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_flipped(self):
self.scatterplot.direction = 'flipped'
self.assertEqual(self.scatterplot.origin, 'top left')
self.assertIsNone(self.scatterplot.x_mapper)
self.assertEqual(self.scatterplot.y_mapper,
self.scatterplot.index_mapper)
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_horizontal_flipped(self):
self.scatterplot.direction = 'flipped'
self.scatterplot.orientation = 'h'
self.assertEqual(self.scatterplot.origin, 'bottom right')
self.assertEqual(self.scatterplot.x_mapper,
self.scatterplot.index_mapper)
self.assertIsNone(self.scatterplot.y_mapper)
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_circle(self):
self.scatterplot.marker = 'circle'
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_custom(self):
path = CompiledPath()
path.move_to(-5, -5)
path.line_to(5, 5)
path.line_to(5, -5)
path.line_to(-5, 5)
path.line_to(-5, -5)
self.scatterplot.marker = 'custom'
self.scatterplot.custom_symbol = path
gc = PlotGraphicsContext(self.size)
gc.render_component(self.scatterplot)
actual = gc.bmp_array[:, :, :]
self.assertFalse(alltrue(actual == 255))
def test_scatter_1d_map_data(self):
points = array([[0, 124.5], [124.5, 0]])
assert_almost_equal(self.scatterplot.map_data(points), array([4.5, 0]))
def test_scatter_1d_map_data_horizontal(self):
self.scatterplot.orientation = 'h'
points = array([[0, 124.5], [124.5, 0]])
assert_almost_equal(self.scatterplot.map_data(points), array([0, 4.5]))
def test_scatter_1d_map_data_flipped(self):
self.scatterplot.direction = 'flipped'
points = array([[0, 124.5], [124.5, 0]])
assert_almost_equal(self.scatterplot.map_data(points),
array([4.5, 9.0]))
def test_scatter_1d_map_data_horizontal_flipped(self):
self.scatterplot.direction = 'flipped'
self.scatterplot.orientation = 'h'
points = array([[0, 124.5], [124.5, 0]])
assert_almost_equal(self.scatterplot.map_data(points),
array([9.0, 4.5]))