class MultiArrayDataTestCase(UnittestTools, unittest.TestCase):
def setUp(self):
self.myarray = arange(20).reshape(10, 2)
self.data_source = MultiArrayDataSource(self.myarray)
def test_init_defaults(self):
data_source = MultiArrayDataSource()
assert_array_equal(data_source._data, empty(shape=(0, 1), dtype=float))
# XXX this doesn't match AbstractDataSource's interface
self.assertEqual(data_source.value_dimension, 1)
self.assertEqual(data_source.sort_order, "ascending")
self.assertFalse(data_source.is_masked())
def test_basic_setup(self):
assert_array_equal(self.myarray, self.data_source._data)
# XXX this doesn't match AbstractDataSource's interface
self.assertEqual(self.data_source.index_dimension, 0)
self.assertEqual(self.data_source.value_dimension, 1)
self.assertEqual(self.data_source.sort_order, "ascending")
self.assertFalse(self.data_source.is_masked())
def test_set_data(self):
new_array = arange(0, 40, 2).reshape(10, 2)
with self.assertTraitChanges(self.data_source, 'data_changed',
count=1):
self.data_source.set_data(new_array)
assert_array_equal(new_array, self.data_source._data)
self.assertEqual(self.data_source.get_bounds(), (0, 38))
self.assertEqual(self.data_source.sort_order, "ascending")
def test_get_data(self):
assert_array_equal(self.myarray, self.data_source.get_data())
def test_get_data_axes(self):
assert_array_equal(arange(0, 20, 2), self.data_source.get_data(axes=0))
def test_get_data_no_data(self):
data_source = MultiArrayDataSource()
assert_array_equal(data_source.get_data(),
empty(shape=(0, 1), dtype=float))
def test_get_data_mask(self):
data, mask = self.data_source.get_data_mask()
assert_array_equal(data, self.myarray)
assert_array_equal(mask, ones(shape=(10, 2), dtype=bool))
def test_bounds(self):
# ascending
bounds = self.data_source.get_bounds()
self.assertEqual(bounds, (0, 19))
# descending
myarray = arange(20)[::-1].reshape(10, 2)
data_source = MultiArrayDataSource(myarray, sort_order='descending')
bounds = data_source.get_bounds()
self.assertEqual(bounds, (0, 19))
# no order
myarray = array([[12, 3], [0, 9], [2, 18], [3, 10]])
data_source = MultiArrayDataSource(myarray, sort_order="none")
bounds = data_source.get_bounds()
self.assertEqual(bounds, (0, 18))
def test_bounds_value(self):
# ascending
bounds = self.data_source.get_bounds(value=0)
self.assertEqual(bounds, (0, 18))
# descending
myarray = arange(20)[::-1].reshape(10, 2)
data_source = MultiArrayDataSource(myarray, sort_order='descending')
bounds = data_source.get_bounds(value=0)
self.assertEqual(bounds, (1, 19))
# no order
myarray = array([[12, 3], [0, 9], [2, 18], [3, 10]])
data_source = MultiArrayDataSource(myarray, sort_order="none")
bounds = data_source.get_bounds(value=0)
self.assertEqual(bounds, (0, 12))
def test_bounds_index(self):
# ascending
bounds = self.data_source.get_bounds(index=0)
self.assertEqual(bounds, (0, 1))
# descending
myarray = arange(20)[::-1].reshape(10, 2)
data_source = MultiArrayDataSource(myarray, sort_order='descending')
bounds = data_source.get_bounds(index=0)
self.assertEqual(bounds, (18, 19))
# no order
myarray = array([[12, 3], [0, 9], [2, 18], [3, 10]])
data_source = MultiArrayDataSource(myarray, sort_order="none")
bounds = data_source.get_bounds(index=0)
self.assertEqual(bounds, (3, 12))
def test_bounds_empty(self):
data_source = MultiArrayDataSource()
bounds = data_source.get_bounds()
# XXX this is sort of inconsistent with test_bounds_all_nans()
self.assertEqual(bounds, (0, 0))
def test_bounds_all_nans(self):
myarray = empty((10, 2))
myarray[:, :] = nan
data_source = MultiArrayDataSource(myarray)
bounds = data_source.get_bounds()
self.assertTrue(isnan(bounds[0]))
self.assertTrue(isnan(bounds[1]))
def test_metadata(self):
self.assertEqual(self.data_source.metadata, {
'annotations': [],
'selections': []
})
@unittest.skip('change handler missing from class')
def test_metadata_changed(self):
with self.assertTraitChanges(self.data_source,
'metadata_changed',
count=1):
self.data_source.metadata = {'new_metadata': True}
@unittest.skip('change handler missing from class')
def test_metadata_items_changed(self):
with self.assertTraitChanges(self.data_source,
'metadata_changed',
count=1):
self.data_source.metadata['new_metadata'] = True
class MultiArrayDataTestCase(UnittestTools, unittest.TestCase):
def setUp(self):
self.myarray = arange(20).reshape(10, 2)
self.data_source = MultiArrayDataSource(self.myarray)
def test_init_defaults(self):
data_source = MultiArrayDataSource()
assert_array_equal(data_source._data, empty(shape=(0, 1), dtype=float))
# XXX this doesn't match AbstractDataSource's interface
self.assertEqual(data_source.value_dimension, 1)
self.assertEqual(data_source.sort_order, "ascending")
self.assertFalse(data_source.is_masked())
def test_basic_setup(self):
assert_array_equal(self.myarray, self.data_source._data)
# XXX this doesn't match AbstractDataSource's interface
self.assertEqual(self.data_source.index_dimension, 0)
self.assertEqual(self.data_source.value_dimension, 1)
self.assertEqual(self.data_source.sort_order, "ascending")
self.assertFalse(self.data_source.is_masked())
def test_set_data(self):
new_array = arange(0, 40, 2).reshape(10, 2)
with self.assertTraitChanges(self.data_source, 'data_changed', count=1):
self.data_source.set_data(new_array)
assert_array_equal(new_array, self.data_source._data)
self.assertEqual(self.data_source.get_bounds(), (0, 38))
self.assertEqual(self.data_source.sort_order, "ascending")
def test_get_data(self):
assert_array_equal(self.myarray, self.data_source.get_data())
def test_get_data_axes(self):
assert_array_equal(arange(0, 20, 2), self.data_source.get_data(axes=0))
def test_get_data_no_data(self):
data_source = MultiArrayDataSource()
assert_array_equal(data_source.get_data(),
empty(shape=(0, 1), dtype=float))
def test_get_data_mask(self):
data, mask = self.data_source.get_data_mask()
assert_array_equal(data, self.myarray)
assert_array_equal(mask, ones(shape=(10, 2), dtype=bool))
def test_bounds(self):
# ascending
bounds = self.data_source.get_bounds()
self.assertEqual(bounds, (0, 19))
# descending
myarray = arange(20)[::-1].reshape(10, 2)
data_source = MultiArrayDataSource(myarray, sort_order='descending')
bounds = data_source.get_bounds()
self.assertEqual(bounds, (0, 19))
# no order
myarray = array([[12, 3], [0, 9], [2, 18], [3, 10]])
data_source = MultiArrayDataSource(myarray, sort_order="none")
bounds = data_source.get_bounds()
self.assertEqual(bounds, (0, 18))
def test_bounds_value(self):
# ascending
bounds = self.data_source.get_bounds(value=0)
self.assertEqual(bounds, (0, 18))
# descending
myarray = arange(20)[::-1].reshape(10, 2)
data_source = MultiArrayDataSource(myarray, sort_order='descending')
bounds = data_source.get_bounds(value=0)
self.assertEqual(bounds, (1, 19))
# no order
myarray = array([[12, 3], [0, 9], [2, 18], [3, 10]])
data_source = MultiArrayDataSource(myarray, sort_order="none")
bounds = data_source.get_bounds(value=0)
self.assertEqual(bounds, (0, 12))
def test_bounds_index(self):
# ascending
bounds = self.data_source.get_bounds(index=0)
self.assertEqual(bounds, (0, 1))
# descending
myarray = arange(20)[::-1].reshape(10, 2)
data_source = MultiArrayDataSource(myarray, sort_order='descending')
bounds = data_source.get_bounds(index=0)
self.assertEqual(bounds, (18, 19))
# no order
myarray = array([[12, 3], [0, 9], [2, 18], [3, 10]])
data_source = MultiArrayDataSource(myarray, sort_order="none")
bounds = data_source.get_bounds(index=0)
self.assertEqual(bounds, (3, 12))
def test_bounds_empty(self):
data_source = MultiArrayDataSource()
bounds = data_source.get_bounds()
# XXX this is sort of inconsistent with test_bounds_all_nans()
self.assertEqual(bounds, (0, 0))
def test_bounds_all_nans(self):
myarray = empty((10, 2))
myarray[:, :] = nan
data_source = MultiArrayDataSource(myarray)
bounds = data_source.get_bounds()
self.assertTrue(isnan(bounds[0]))
self.assertTrue(isnan(bounds[1]))
def test_metadata(self):
self.assertEqual(self.data_source.metadata,
{'annotations': [], 'selections': []})
@unittest.skip('change handler missing from class')
def test_metadata_changed(self):
with self.assertTraitChanges(self.data_source, 'metadata_changed', count=1):
self.data_source.metadata = {'new_metadata': True}
@unittest.skip('change handler missing from class')
def test_metadata_items_changed(self):
with self.assertTraitChanges(self.data_source, 'metadata_changed', count=1):
self.data_source.metadata['new_metadata'] = True
