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.reset_model(self.data, self.index_ids, ('test1', 'test2'), (),
('test3', ), (5, ))
index_set = sorted(set([('d', 'dd'), ('e', 'ee')]))
index_header_values = model._get_unique_index_values(
('test1', 'test2'))
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.reset_model(self.data, self.index_ids)
index_set = sorted({('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)
def test_reset_model(self):
"""test reset model data"""
row_headers = [('a', 'aa', 1), ('a', 'bb', 2), ('b', 'cc', 3),
('c', 'cc', 4), ('d', 'dd', 5), ('e', 'ee', 5)]
column_headers = []
model = PivotModel()
model.reset_model(self.data, self.index_ids)
self.assertEqual(model._data, self.data)
self.assertEqual(model.index_ids, tuple(self.index_ids))
self.assertEqual(model.pivot_rows, tuple(self.index_ids))
self.assertEqual(model.pivot_columns, ())
self.assertEqual(model.pivot_frozen, ())
self.assertEqual(model.frozen_value, ())
self.assertEqual(model._row_data_header, row_headers)
self.assertEqual(model._column_data_header, column_headers)
def test_get_pivoted_data1(self):
"""get data with pivot and frozen index and tuple_index_entries"""
model = PivotModel()
model.reset_model(self.data, self.index_ids)
model.set_pivot(['test2'], ['test3'], ['test1'], ('a', ))
data = [['value_a_aa_1', None], [None, 'value_a_bb_2']]
data_model = [[d for d in inner]
for inner in model.get_pivoted_data(range(2), range(2))]
self.assertEqual(data_model, data)
def test_set_pivot_with_frozen(self):
"""Test set_pivot with frozen dimension"""
model = PivotModel()
model.reset_model(self.data, self.index_ids)
model.set_pivot(['test2'], ['test3'], ['test1'], ('a', ))
row_headers = [('aa', ), ('bb', )]
data = [['value_a_aa_1', None], [None, 'value_a_bb_2']]
column_headers = [(1, ), (2, )]
data_model = [[d for d in inner]
for inner in model.get_pivoted_data(range(2), range(2))]
self.assertEqual(model._row_data_header, row_headers)
self.assertEqual(model._column_data_header, column_headers)
self.assertEqual(data_model, data)
def test_reset_model_with_pivot(self):
"""Test set data with pivot and tuple_index_entries"""
column_headers = [
('a', 'aa', 1),
('a', 'bb', 2),
('b', 'cc', 3),
('c', 'cc', 4),
('d', 'dd', 5),
('e', 'ee', 5),
]
row_headers = []
model = PivotModel()
model.reset_model(self.data,
self.index_ids,
rows=(),
columns=tuple(self.index_ids))
self.assertEqual(model._data, self.data)
self.assertEqual(model.index_ids, tuple(self.index_ids))
self.assertEqual(model.pivot_rows, ())
self.assertEqual(model.pivot_columns, tuple(self.index_ids))
self.assertEqual(model.pivot_frozen, ())
self.assertEqual(model.frozen_value, ())
self.assertEqual(model._row_data_header, row_headers)
self.assertEqual(model._column_data_header, column_headers)
def test_set_pivot(self):
"""Test set_pivot"""
model = PivotModel()
model.reset_model(self.data, self.index_ids)
model.set_pivot(['test1', 'test2'], ['test3'], [], ())
row_headers = [('a', 'aa'), ('a', 'bb'), ('b', 'cc'), ('c', 'cc'),
('d', 'dd'), ('e', 'ee')]
column_headers = [(1, ), (2, ), (3, ), (4, ), (5, )]
self.assertEqual(model._row_data_header, row_headers)
self.assertEqual(model._column_data_header, column_headers)
def test_get_pivoted_data3(self):
"""get data from pivoted model"""
model = PivotModel()
model.reset_model(self.data, self.index_ids)
model.set_pivot(['test1', 'test2'], ['test3'], [], ())
data = [
['value_a_aa_1', None, None, None, None],
[None, 'value_a_bb_2', None, None, None],
[None, None, 'value_b_cc_3', None, None],
[None, None, None, 'value_c_cc_4', None],
[None, None, None, None, 'value_d_dd_5'],
[None, None, None, None, 'value_e_ee_5'],
]
data_model = [[d for d in inner]
for inner in model.get_pivoted_data(range(6), range(5))]
self.assertEqual(data_model, data)
def test_init_model(self):
PivotModel()