def test_get_value_exception(self):
o = SimpleDemoModel()
ds = SimpleDemoModelSource.as_datasource()
col = ColumnCallable(FakeCallable)
with self.assertRaises(ValueError):
col.get_value(o, ds)
def test_custom_column_callable(self):
G(SimpleDemoModel, n=2, char='abc', integer1=10, integer2=20)
def _custom_callable(obj, ds):
return obj.integer1 + 100
ds = Datasource.as_datasource(model=SimpleDemoModel,
columns=[
Column('integer1'),
Column('integer2'),
ColumnCallable(_custom_callable)
])
self.assertSequenceEqual(ds, [(10, 20, 110), (10, 20, 110)])
def test_get_value(self):
col = ColumnCallable(FakeCallable)
v = col.get_value(None, None)
self.assertIsInstance(v, RowValue)
def test_init(self):
col = ColumnCallable(FakeCallable)
self.assertEqual(col.title, 'Callable')
