def setUp(self): self.context = GeoContext() self.a = array([1, 2, 3]) self.context['a'] = self.a self.b = array([4, 5, 6, 7]) self.context['b'] = self.b self.datasource = DataContextDataSource(context=self.context, context_name='a')
def setUp(self): self.context = GeoContext() self.a = array([1, 2, 3]) self.context["a"] = self.a self.b = array([4, 5, 6, 7]) self.context["b"] = self.b self.datasource = DataContextDataSource(context=self.context, context_name="a")
class GeoContextDataSourceBasicTest(unittest.TestCase): def setUp(self): self.context = GeoContext() self.a = array([1, 2, 3]) self.context["a"] = self.a self.b = array([4, 5, 6, 7]) self.context["b"] = self.b self.datasource = DataContextDataSource(context=self.context, context_name="a") def test_get_data(self): self.assertTrue((self.datasource.get_data() == self.a).all()) return def test_length(self): self.assertEqual(self.datasource.get_size(), len(self.a)) return def test_event(self): self.did_fire = False def handle_fire(): self.did_fire = True return self.datasource.on_trait_change(handle_fire, "data_changed") self.context["a"] = array([7, 8, 9]) self.assertTrue(self.did_fire) self.datasource.on_trait_change(handle_fire, "data_changed", remove=True) return
class GeoContextDataSourceBasicTest(unittest.TestCase): def setUp(self): self.context = GeoContext() self.a = array([1, 2, 3]) self.context['a'] = self.a self.b = array([4, 5, 6, 7]) self.context['b'] = self.b self.datasource = DataContextDataSource(context=self.context, context_name='a') def test_get_data(self): self.assertTrue((self.datasource.get_data() == self.a).all()) return def test_length(self): self.assertEqual(self.datasource.get_size(), len(self.a)) return def test_event(self): self.did_fire = False def handle_fire(): self.did_fire = True return self.datasource.on_trait_change(handle_fire, 'data_changed') self.context['a'] = array([7, 8, 9]) self.assertTrue(self.did_fire) self.datasource.on_trait_change(handle_fire, 'data_changed', remove=True) return
def get_datasource(self, name): """Gets a chaco DataSource for the named object.""" return DataContextDataSource(context=self.context, context_name=name)