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)