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
|
import unittest
from enthought.pyface.grid.api \
import GridRow, GridColumn, SimpleGridModel
class CompositeGridModelTestCase( unittest.TestCase ):
def setUp(self):
self.model = SimpleGridModel(data=[[None,2],[3,4]],
rows=[GridRow(label='foo'),
GridRow(label='bar')],
columns=[GridColumn(label='cfoo'),
GridColumn(label='cbar')]
)
return
def test_get_column_count(self):
self.assertEqual(self.model.get_column_count(), 2)
return
def test_get_row_count(self):
self.assertEqual(self.model.get_row_count(), 2)
return
def test_get_row_name(self):
# Regardless of the rows specified in the composed models, the
# composite model returns its own rows.
self.assertEqual(self.model.get_row_name(0), 'foo')
self.assertEqual(self.model.get_row_name(1), 'bar')
return
def test_get_column_name(self):
self.assertEqual(self.model.get_column_name(0), 'cfoo')
self.assertEqual(self.model.get_column_name(1), 'cbar')
return
def test_get_value(self):
self.assertEqual(self.model.get_value(0,0), None)
self.assertEqual(self.model.get_value(0,1), 2)
self.assertEqual(self.model.get_value(1,0), 3)
self.assertEqual(self.model.get_value(1,1), 4)
return
def test_is_cell_empty(self):
rows = self.model.get_row_count()
columns = self.model.get_column_count()
self.assertEqual(self.model.is_cell_empty(0,0), True,
"Cell containing None should be empty.")
self.assertEqual(self.model.is_cell_empty(10,10), True,
"Cell outside the range of values should be empty.")
return
#### EOF ######################################################################
|