def test_add_entries(self):
class DummyCalculation(object):
name = 'dummy_calc'
needed_columns = ['foo', 'bar']
ColumnCalculator.register_calculation(DummyCalculation)
cols = ['braz', 'uiae', 'dummy_calc', 'nrtd']
cc = ColumnCalculator(cols)
expected = ['braz', 'uiae', 'foo', 'bar', 'nrtd']
result = cc.add_entries(cols)
self.assertEqual(result, expected)
def test_calculate_entries(self):
class DummyCalculation(object):
name = 'dummy_calc'
needed_columns = ['foo', 'bar']
@staticmethod
def calc(val1, val2):
return val1 - val2
ColumnCalculator.register_calculation(DummyCalculation)
cols = ['braz', 'uiae', 'dummy_calc', 'nrtd']
cc = ColumnCalculator(cols)
entries = ['braz', 'uiae', 1, 5, 'nrtd']
result = cc.calculate_entries(entries)
expected = ('braz', 'uiae', -4, 'nrtd', )
self.assertEqual(result, expected)
