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)