def test_empty(self): ds = GridDataSource() self.assert_(ds.sort_order == ('none', 'none')) self.assert_(ds.index_dimension == 'image') self.assert_(ds.value_dimension == 'scalar') self.assert_(ds.metadata == {"selections":[], "annotations":[]}) xdata, ydata = ds.get_data() assert_ary_(xdata.get_data(), array([])) assert_ary_(ydata.get_data(), array([])) self.assert_(ds.get_bounds() == ((0,0),(0,0)))
def test_init(self): test_xd = array([1,2,3]) test_yd = array([1.5, 0.5, -0.5, -1.5]) test_sort_order = ('ascending', 'descending') ds = GridDataSource(xdata=test_xd, ydata=test_yd, sort_order=test_sort_order) self.assert_(ds.sort_order == test_sort_order) xd, yd = ds.get_data() assert_ary_(xd.get_data(), test_xd) assert_ary_(yd.get_data(), test_yd) self.assert_(ds.get_bounds() == ((min(test_xd),min(test_yd)), (max(test_xd),max(test_yd))))