def test_pivot(): table1 = (('region', 'gender', 'style', 'units'), ('east', 'boy', 'tee', 12), ('east', 'boy', 'golf', 14), ('east', 'boy', 'fancy', 7), ('east', 'girl', 'tee', 3), ('east', 'girl', 'golf', 8), ('east', 'girl', 'fancy', 18), ('west', 'boy', 'tee', 12), ('west', 'boy', 'golf', 15), ('west', 'boy', 'fancy', 8), ('west', 'girl', 'tee', 6), ('west', 'girl', 'golf', 16), ('west', 'girl', 'fancy', 1)) table2 = pivot(table1, 'region', 'gender', 'units', sum) expect2 = (('region', 'boy', 'girl'), ('east', 33, 29), ('west', 35, 23)) ieq(expect2, table2) ieq(expect2, table2)
def test_pivot_empty(): table1 = (('region', 'gender', 'style', 'units'),) table2 = pivot(table1, 'region', 'gender', 'units', sum) expect2 = (('region',),) ieq(expect2, table2)