1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
from chaco.api import Plot, ArrayPlotData
from traits.api import HasTraits, Instance
from enable.component_editor import ComponentEditor
from traitsui.api import Item, View
import numpy as np
import nose
from _tools import store_exceptions_on_all_threads, assert_raises
class PlotViewer(HasTraits):
plot = Instance(Plot)
traits_view = View(Item('plot', editor=ComponentEditor()))
def test_bounds_2d_case():
# test for bug: contour and image plots should support the case where
# xbounds and ybounds are 2d arrays resulting from meshgrids
xs = np.linspace(-10,10,200)
ys = np.linspace(-10,10,400)
x, y = np.meshgrid(xs,ys)
z = x + y
plotdata = ArrayPlotData()
plotdata.set_data("z", z)
plot = Plot(plotdata)
plot.contour_plot("z", xbounds=x, ybounds=y)
# try to display it, that's when the exception is raised
with store_exceptions_on_all_threads():
pv = PlotViewer(plot=plot)
pv.edit_traits()
def test_process_2d_bounds():
# behavior: _process_2d_bounds accepts all possible ways to set x and y
# bounds in 2d plots and returns a 1d array with equally spaced
# intervals between the lower and upper bound of the data. The number
# of elements in the 1d array must be of one element larger than the
# shape of the data, because it includes the upper bound.
height, width = 20, 10
array_data = np.ones(shape=(height, width))
plot = Plot()
# bounds is None : infer from array_data shape
xs = plot._process_2d_bounds(None, array_data, 1)
assert xs.shape[0] == width + 1
ys = plot._process_2d_bounds(None, array_data, 0)
assert ys.shape[0] == height + 1
# bounds is a tuple : it defines lower and upper range
bounds = (1.0, 100.0)
xs = plot._process_2d_bounds(bounds, array_data, 1)
assert xs.shape[0] == width + 1
assert xs[0] == bounds[0] and xs[-1] == bounds[1]
# bounds is a 1D array: the first and last elements are used to create
# equally spaced intervals. Bounds must be of one element larger than the
# corresponding axis in array_data, or it will raise a Value error
bounds = np.zeros((height+1, ))
bounds[0], bounds[-1] = 0.2, 21.3
ys = plot._process_2d_bounds(bounds, array_data, 0)
assert ys.shape[0] == height + 1
assert ys[0] == bounds[0] and ys[-1] == bounds[-1]
with assert_raises(ValueError):
bounds = np.zeros((width // 2, ))
plot._process_2d_bounds(bounds, array_data, 0)
# bounds is a 2D array: the first and last elements along the appropriate
# axis are used to create equally spaced intervals.
# The size of the bounds must be the same as the data array, or this
# sill raise a ValueError
xbounds, ybounds = np.meshgrid(np.arange(width), np.arange(height))
xs = plot._process_2d_bounds(xbounds, array_data, 1)
assert xs.shape[0] == width + 1
assert xs[0] == xbounds[0,0] and xs[-2] == xbounds[0,-1]
with assert_raises(ValueError):
plot._process_2d_bounds(xbounds[:5,:], array_data, 1)
ys = plot._process_2d_bounds(ybounds, array_data, 0)
assert ys.shape[0] == height + 1
assert ys[0] == ybounds[0,0] and ys[-2] == ybounds[-1,0]
if __name__ == '__main__':
nose.main()
|
