def test_empty(self):
table = pivots.pivot_table_from_select(
self.mytable.select(),
rows='yearmonth', cols='gender', values='price'
)
self.assertIsNone(table)
def test_empty(self):
table = pivots.pivot_table_from_select(self.mytable.select(),
rows='yearmonth',
cols='gender',
values='price')
self.assertIsNone(table)
def test_len(self):
self._insert_data(1, '201001', 'male', 10)
self._insert_data(2, '201001', 'male', 5)
self._insert_data(3, '201001', 'female', 10)
table = pivots.pivot_table_from_select(self.mytable.select(),
rows='yearmonth',
cols='gender',
values=(len, 'price'))
self.assertEquals(1, table.price.female['201001'])
self.assertEquals(2, table.price.male['201001'])
def test_len(self):
self._insert_data(1, '201001', 'male', 10)
self._insert_data(2, '201001', 'male', 5)
self._insert_data(3, '201001', 'female', 10)
table = pivots.pivot_table_from_select(
self.mytable.select(),
rows='yearmonth', cols='gender', values=(len, 'price')
)
self.assertEquals(1, table.price.female['201001'])
self.assertEquals(2, table.price.male['201001'])
def test_multiple_columns(self):
self._insert_data(1, '201001', 'male', 10, town='BA')
self._insert_data(2, '201001', 'male', 5, town='KE')
self._insert_data(3, '201001', 'female', 10, town='BA')
table = pivots.pivot_table_from_select(self.mytable.select(),
rows='yearmonth',
cols=['town', 'gender'],
values=(len, 'price'))
self.assertEquals(1, table.price.BA.female['201001'])
self.assertEquals(1, table.price.BA.male['201001'])
self.assertEquals(1, table.price.KE.male['201001'])
def test_multiple_columns(self):
self._insert_data(1, '201001', 'male', 10, town='BA')
self._insert_data(2, '201001', 'male', 5, town='KE')
self._insert_data(3, '201001', 'female', 10, town='BA')
table = pivots.pivot_table_from_select(
self.mytable.select(),
rows='yearmonth', cols=['town', 'gender'], values=(len, 'price')
)
self.assertEquals(1, table.price.BA.female['201001'])
self.assertEquals(1, table.price.BA.male['201001'])
self.assertEquals(1, table.price.KE.male['201001'])
def test_pivot_table(self):
self._insert_data(1, '201001', 'male', 10)
self._insert_data(2, '201001', 'male', 5)
self._insert_data(3, '201001', 'female', 10)
select = self.mytable.select().order_by('yearmonth', 'gender')
table = pivots.pivot_table_from_select(
select,
rows='yearmonth', cols='gender', values='price'
)
self.assertEquals(' price \n'
'gender female male\n'
'yearmonth \n'
'201001 10 15', str(table))
def test_pivot_table(self):
self._insert_data(1, '201001', 'male', 10)
self._insert_data(2, '201001', 'male', 5)
self._insert_data(3, '201001', 'female', 10)
select = self.mytable.select().order_by('yearmonth', 'gender')
table = pivots.pivot_table_from_select(select,
rows='yearmonth',
cols='gender',
values='price')
self.assertEquals(
' price \n'
'gender female male\n'
'yearmonth \n'
'201001 10 15', str(table))
def test_multiple_functions(self):
self._insert_data(1, '201001', 'male', 10)
self._insert_data(2, '201001', 'male', 5)
self._insert_data(3, '201001', 'female', 10)
table = pivots.pivot_table_from_select(self.mytable.select(),
rows='yearmonth',
cols='gender',
values=[('count', 'price'),
('sum', 'price')])
self.assertEquals(2, table.price_count.male['201001'])
self.assertEquals(1, table.price_count.female['201001'])
self.assertEquals(15, table.price_sum.male['201001'])
self.assertEquals(10, table.price_sum.female['201001'])
def test_multiple_functions(self):
self._insert_data(1, '201001', 'male', 10)
self._insert_data(2, '201001', 'male', 5)
self._insert_data(3, '201001', 'female', 10)
table = pivots.pivot_table_from_select(
self.mytable.select(),
rows='yearmonth', cols='gender',
values=[('count', 'price'), ('sum', 'price')]
)
self.assertEquals(2, table.price_count.male['201001'])
self.assertEquals(1, table.price_count.female['201001'])
self.assertEquals(15, table.price_sum.male['201001'])
self.assertEquals(10, table.price_sum.female['201001'])
