def test_get_unique_index_values2(self): """test that _get_unique_index_values returns unique values for specified indexes with filter index and value""" model = PivotModel() model.set_new_data(self.data, self.index_names, self.index_types) index_set = set([('d', 'dd'), ('e', 'ee')]) index_header_values = model._get_unique_index_values(('test1','test2'), ('test3',), (5,)) self.assertEqual(index_header_values, index_set)
def test_get_unique_index_values1(self): """test that _get_unique_index_values returns unique values for specified indexes""" model = PivotModel() model.set_new_data(self.data, self.index_names, self.index_types) index_set = set([('a', 'aa'), ('a', 'bb'), ('b', 'cc'), ('c', 'cc'), ('d', 'dd'), ('e', 'ee')]) index_header_values = model._get_unique_index_values(('test1','test2'), (), ()) self.assertEqual(index_header_values, index_set)