def testSetting(self):
class DataObject:
def __init__(self):
self.isChecked = True
data = DataObject()
col = ColumnDefn(checkStateGetter="isChecked")
self.assertEqual(col.GetCheckState(data), True)
col.SetCheckState(data, None)
self.assertEqual(col.GetCheckState(data), None)
col.SetCheckState(data, False)
self.assertEqual(col.GetCheckState(data), False)
def testGetting(self):
class DataObject:
def __init__(self):
self.isChecked = True
def ShouldBeChecked(self):
return False
data = DataObject()
col = ColumnDefn(checkStateGetter="isChecked")
self.assertEqual(col.GetCheckState(data), True)
col = ColumnDefn(checkStateGetter="ShouldBeChecked")
self.assertEqual(col.GetCheckState(data), False)
