class TestTableCompute(unittest.TestCase):
def setUp(self):
self.rows = (
('a', 2, 3, 4),
(None, 3, 5, None),
('a', 2, 4, None),
('b', 3, 4, None)
)
self.number_type = NumberType()
self.text_type = TextType()
self.column_types = (self.text_type, self.number_type, self.number_type, self.number_type)
self.column_names = ('one', 'two', 'three', 'four')
self.table = Table(self.rows, self.column_types, self.column_names)
def test_compute(self):
new_table = self.table.compute('test', self.number_type, lambda r: r['two'] + r['three'])
self.assertIsNot(new_table, self.table)
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
self.assertSequenceEqual(new_table.rows[0], ('a', 2, 3, 4, 5))
self.assertSequenceEqual(new_table.columns['test'], (5, 8, 6, 7))
def test_percent_change(self):
new_table = self.table.percent_change('two', 'three', 'test')
self.assertIsNot(new_table, self.table)
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
to_one_place = lambda d: d.quantize(Decimal('0.1'))
self.assertSequenceEqual(new_table.rows[0], ('a', Decimal('2'), Decimal('3'), Decimal('4'), Decimal('50.0')))
self.assertEqual(to_one_place(new_table.columns['test'][0]), Decimal('50.0'))
self.assertEqual(to_one_place(new_table.columns['test'][1]), Decimal('66.7'))
self.assertEqual(to_one_place(new_table.columns['test'][2]), Decimal('100.0'))
self.assertEqual(to_one_place(new_table.columns['test'][3]), Decimal('33.3'))
def test_rank(self):
new_table = self.table.rank(lambda r: r['two'], 'rank')
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
self.assertSequenceEqual(new_table.rows[0], ('a', 2, 3, 4, 1))
self.assertSequenceEqual(new_table.rows[1], (None, 3, 5, None, 3))
self.assertSequenceEqual(new_table.rows[2], ('a', 2, 4, None, 1))
self.assertSequenceEqual(new_table.rows[3], ('b', 3, 4, None, 3))
self.assertSequenceEqual(new_table.columns['rank'], (1, 3, 1, 3))
def test_rank2(self):
new_table = self.table.rank(lambda r: r['one'], 'rank')
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
self.assertSequenceEqual(new_table.rows[0], ('a', 2, 3, 4, 1))
self.assertSequenceEqual(new_table.rows[1], (None, 3, 5, None, 4))
self.assertSequenceEqual(new_table.rows[2], ('a', 2, 4, None, 1))
self.assertSequenceEqual(new_table.rows[3], ('b', 3, 4, None, 3))
self.assertSequenceEqual(new_table.columns['rank'], (1, 4, 1, 3))
def test_rank_column_name(self):
new_table = self.table.rank('two', 'rank')
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
self.assertSequenceEqual(new_table.rows[0], ('a', 2, 3, 4, 1))
self.assertSequenceEqual(new_table.rows[1], (None, 3, 5, None, 3))
self.assertSequenceEqual(new_table.rows[2], ('a', 2, 4, None, 1))
self.assertSequenceEqual(new_table.rows[3], ('b', 3, 4, None, 3))
self.assertSequenceEqual(new_table.columns['rank'], (1, 3, 1, 3))
class TestTableCompute(unittest.TestCase):
def setUp(self):
self.rows = (('a', 2, 3, 4), (None, 3, 5, None), ('a', 2, 4, None),
('b', 3, 4, None))
self.number_type = NumberType()
self.text_type = TextType()
self.column_types = (self.text_type, self.number_type,
self.number_type, self.number_type)
self.column_names = ('one', 'two', 'three', 'four')
self.table = Table(self.rows, self.column_types, self.column_names)
def test_compute(self):
new_table = self.table.compute('test', self.number_type,
lambda r: r['two'] + r['three'])
self.assertIsNot(new_table, self.table)
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
self.assertSequenceEqual(new_table.rows[0], ('a', 2, 3, 4, 5))
self.assertSequenceEqual(new_table.columns['test'], (5, 8, 6, 7))
def test_percent_change(self):
new_table = self.table.percent_change('two', 'three', 'test')
self.assertIsNot(new_table, self.table)
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
to_one_place = lambda d: d.quantize(Decimal('0.1'))
self.assertSequenceEqual(
new_table.rows[0],
('a', Decimal('2'), Decimal('3'), Decimal('4'), Decimal('50.0')))
self.assertEqual(to_one_place(new_table.columns['test'][0]),
Decimal('50.0'))
self.assertEqual(to_one_place(new_table.columns['test'][1]),
Decimal('66.7'))
self.assertEqual(to_one_place(new_table.columns['test'][2]),
Decimal('100.0'))
self.assertEqual(to_one_place(new_table.columns['test'][3]),
Decimal('33.3'))
def test_rank(self):
new_table = self.table.rank(lambda r: r['two'], 'rank')
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
self.assertSequenceEqual(new_table.rows[0], ('a', 2, 3, 4, 1))
self.assertSequenceEqual(new_table.rows[1], (None, 3, 5, None, 3))
self.assertSequenceEqual(new_table.rows[2], ('a', 2, 4, None, 1))
self.assertSequenceEqual(new_table.rows[3], ('b', 3, 4, None, 3))
self.assertSequenceEqual(new_table.columns['rank'], (1, 3, 1, 3))
def test_rank2(self):
new_table = self.table.rank(lambda r: r['one'], 'rank')
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
self.assertSequenceEqual(new_table.rows[0], ('a', 2, 3, 4, 1))
self.assertSequenceEqual(new_table.rows[1], (None, 3, 5, None, 4))
self.assertSequenceEqual(new_table.rows[2], ('a', 2, 4, None, 1))
self.assertSequenceEqual(new_table.rows[3], ('b', 3, 4, None, 3))
self.assertSequenceEqual(new_table.columns['rank'], (1, 4, 1, 3))
def test_rank_column_name(self):
new_table = self.table.rank('two', 'rank')
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
self.assertSequenceEqual(new_table.rows[0], ('a', 2, 3, 4, 1))
self.assertSequenceEqual(new_table.rows[1], (None, 3, 5, None, 3))
self.assertSequenceEqual(new_table.rows[2], ('a', 2, 4, None, 1))
self.assertSequenceEqual(new_table.rows[3], ('b', 3, 4, None, 3))
self.assertSequenceEqual(new_table.columns['rank'], (1, 3, 1, 3))
def test_z_scores(self):
new_table = self.table.z_scores('two', 'z-scores')
self.assertEqual(len(new_table.rows), 4)
self.assertEqual(len(new_table.columns), 5)
self.assertSequenceEqual(new_table.rows[0], ('a', 2, 3, 4, -1))
self.assertSequenceEqual(new_table.rows[1], (None, 3, 5, None, 1))
self.assertSequenceEqual(new_table.rows[2], ('a', 2, 4, None, -1))
self.assertSequenceEqual(new_table.rows[3], ('b', 3, 4, None, 1))
self.assertSequenceEqual(new_table.columns['z-scores'], (-1, 1, -1, 1))
