def test_update_matrix_with_labels_and_cells(self):
"""simulate the update of the matrix with new formatted labels for
rows and column items
"""
from format_labels import FormatSettings
settings = FormatSettings(label_format="{0.Group} - {0}")
row_expected = [
row.Member.Group.Label + " - " + row.Member.Label for row in self.m
]
col_expected = [
col.TopMember.Group.Label + " - " + col.TopMember.Label
for col in self.m[0]
]
for row in self.m:
row.Member.Label = settings.label_format(row.Member)
for col in self.m[0]:
col.TopMember.Label = settings.label_format(col.TopMember)
row_labels = [row.Member.Label for row in self.m]
col_labels = [col.TopMember.Label for col in self.m[0]]
self.assertEqual(row_labels, row_expected)
self.assertEqual(col_labels, col_expected)
def test_fill_defaults(self):
""" simulates what might happen during a fill with default arguments"""
from format_labels import FormatSettings
settings = FormatSettings()
#matrix
label = settings.label_format(self.m)
##- This fails currently as matrix has no str() function set on it
##- instead it returns the object.
#####--------->
#self.assertEqual(label, self.m)
#rows
for child in self.m:
label = settings.label_format(child.Member)
self.assertEqual(label, child.Member.Label)
#cols
for child in self.m[0]:
label = settings.label_format(child.TopMember)
self.assertEqual(label, child.TopMember.Label)
#side group
for group in self.m.SideAxis.Groups:
label = settings.label_format(group)
self.assertEqual(label, group.Label)
#top group
for group in self.m.TopAxis.Groups:
label = settings.label_format(group)
self.assertEqual(label, group.Label)
def test_fill_defaults(self):
""" simulates being default settings for different labels within object"""
from format_labels import FormatSettings
settings = FormatSettings()
#top level label
label = settings.label_format(self.test_axis)
self.assertEqual(label,self.test_axis.Label)
#children
for child in self.test_axis.Children:
label = settings.label_format(child)
self.assertEqual(label,child.Label)
#2nd level children
for child in self.test_axis.Children:
for item in child.Children:
label = settings.label_format(item)
self.assertEqual(label,item.Label)
#individual items
label = settings.label_format(self.test_axis.Children[0])
self.assertEqual(label,self.test_axis.Children[0].Label)
label = settings.label_format(self.test_axis.Children[0].Children[0])
self.assertEqual(label,self.test_axis.Children[0].Children[0].Label)
def test_fill_label_and_cell_format(self):
""" simulates a fill with a string format for row and cell items passed as
argument
"""
from format_labels import FormatSettings
#cell item 0
settings = FormatSettings(
label_format=
"{0.SideMember.Group} - {0.SideMember} - {0.TopMember.Group} - {0.TopMember} - (n = {0[0].Value})",
cell_format="{0.SideMember} - {0[0].Value}")
for row in self.m:
label = settings.label_format(row[0])
expected = row.Member.Group.Label + " - " + row.Member.Label + " - " + row[
0].TopMember.Group.Label + " - " + row[
0].TopMember.Label + " - (n = " + row[0][0].Value + ")"
self.assertEqual(expected, label)
#cell item 1
settings = FormatSettings(
label_format=
"{0.SideMember.Group} - {0.SideMember} - {0.TopMember.Group} - {0.TopMember} - (n = {0[0].Value})",
cell_format="{0.SideMember} - {0[0].Value}")
for row in self.m:
for col in row:
label2 = settings.cell_format(col)
expected2 = col.SideMember.Label + " - " + col[0].Value
self.assertEqual(expected2, label2)
def test_fill_simple_format(self):
""" simulates what might happen during a fill with a string format
given as an argument
"""
from format_labels import FormatSettings
settings = FormatSettings(label_format="{0.Group};{0}")
#rows
for child in self.m:
label = settings.label_format(child.Member)
expected = child.Member.Group.Label + ";" + child.Member.Label
self.assertEqual(expected, label)
#cols
for child in self.m[0]:
label = settings.label_format(child.TopMember)
expected = child.TopMember.Group.Label + ";" + child.TopMember.Label
self.assertEqual(expected, label)
def test_replacement_func(self):
""" simulates what might happen in a fill if a custom function is used
to format arguments, for example to replace texts.
"""
from format_labels import FormatSettings
corrections = {"Male": "m",
"Female": "f",
"Child": "ch"}
def lookup_label_format( item ):
""" run the replacement texts on the string"""
#make sure its a string
item = str(item)
for key in corrections.keys():
item = item.replace(key,corrections[key])
return item
settings = FormatSettings(label_format = lookup_label_format)
#children
for child in self.test_axis.Children:
label = settings.label_format(child)
if child.Label == "Male":
expected = "m"
if child.Label == "Female":
expected = "f"
self.assertEqual(expected,label)
#2nd level child
for child in self.test_axis.Children:
for item in child.Children:
label = settings.label_format(item)
if "Child" in item.Label:
expected = "ch " + label.split("ch ")[1]
self.assertEqual(expected,label)
def test_fill_simple_format(self):
""" simulates what might happen during a
fill with a string format given as an argument
"""
from format_labels import FormatSettings
settings = FormatSettings(label_format="{0.Parent};{0}")
#children
for child in self.test_axis.Children:
label = settings.label_format(child)
expected = child.Parent.Label + ";" + child.Label
self.assertEqual(expected,label)
#2nd level children
for child in self.test_axis.Children:
for item in child.Children:
label = settings.label_format(item)
expected = item.Parent.Label + ";" + item.Label
self.assertEqual(expected,label)
def test_update_matrix_with_labels_and_cells(self):
"""simulate the update of the matrix with new formatted labels for
rows and cell items
"""
from format_labels import FormatSettings
settings = FormatSettings(
label_format="{0.SideMember.Group} - {0[0].Value}",
cell_format="{0.SideMember} - {0[0].Value}")
init_row = self.m[0].Member.Label
init_col = self.m[0].TopMember.Label
init_cell = self.m[0][0][0].Value
init_cell2 = self.m[0][0][1].Value
#update the cell values to "{0.SideMember} - {0[0].Value}"
settings = FormatSettings(
label_format="{0.SideMember.Group} - {0[0].Value}",
cell_format="{0.SideMember} - {0[0].Value}")
for row in self.m:
for col in row:
col[0].Value = settings.cell_format(col)
settings = FormatSettings(
label_format="{0.SideMember.Group} - {0[0].Value}",
cell_format="{0.TopMember} - {0[1].Value}")
for row in self.m:
for col in row:
col[1].Value = settings.cell_format(col)
#update the labels to "{0.SideMember.Group} - {0[0].Value}"
settings = FormatSettings(
label_format="{0.SideMember.Group} - {0[0].Value}",
cell_format="{0.SideMember} - {0[0].Value}")
for row in self.m:
row.Member.Label = settings.label_format(row[0])
labels = [row.Member.Label for row in self.m]
expected = [
row[0].SideMember.Group.Label + " - " + row[0][0].Value
for row in self.m
]
self.assertEqual(labels, expected)
self.assertEqual(self.m[0][0][0].Value, init_row + " - " + init_cell)
self.assertEqual(self.m[0][0][1].Value, init_col + " - " + init_cell2)
def test_fill_cell_defaults(self):
""" simulates being default settings for different cell values
settings = FormatSettings(cell_format = "{0}")
"""
from format_labels import FormatSettings
settings = FormatSettings(cell_format = "{0}")
#children
for child in self.test_axis.Children:
for val in child.Values:
label = settings.cell_format(val)
self.assertEqual(label,str(val))
#2nd level children
for child in self.test_axis.Children:
for item in child.Children:
for val in item.Values:
label = settings.label_format(val)
self.assertEqual(label,str(val))
def test_replacement_func(self):
"""simulates what might happen in a fill if a custom function is used
to format arguments
"""
from format_labels import FormatSettings
corrections = {
"RowGroup": "Row Group",
"myRow": "My Row",
"ColumnGroup": "Column Group",
"myColumn": "My Column"
}
def lookup_label_format(item):
"""run the replacement texts on the string"""
#make sure item is a string
item = str(item)
for key in corrections.keys():
item = item.replace(key, corrections[key])
return item
settings = FormatSettings(label_format=lookup_label_format
) # pass a function to the format simulates
settings2 = FormatSettings(label_format="{0.Group}: {0}")
#rows
for row in self.m:
label = settings2.label_format(row.Member)
label = settings.label_format(label) # replace existing texts
expected = (row.Member.Group.Label + ": " +
row.Member.Label).replace("RowGroup",
"Row Group").replace(
"myRow", "My Row")
self.assertEqual(expected, label)
#cols
for col in self.m[0]:
label = settings2.label_format(col.TopMember)
label = settings.label_format(label) # replace existing texts
expected = (col.TopMember.Group.Label +
": " + col.TopMember.Label).replace(
"ColumnGroup",
"Column Group").replace("myColumn", "My Column")
self.assertEqual(expected, label)
settings2 = FormatSettings(label_format="{0};{0.Count}")
#side groups
for group in self.m.SideAxis.Groups:
label = settings2.label_format(group)
label = settings.label_format(label) # replace existing texts
expected = (group.Label + ";" + str(group.Count)).replace(
"RowGroup", "Row Group").replace("myRow", "My Row")
self.assertEqual(expected, label)
#top groups
for group in self.m.TopAxis.Groups:
label = settings2.label_format(group)
label = settings.label_format(label) # replace existing texts
expected = (group.Label + ";" + str(group.Count)).replace(
"ColumnGroup", "Column Group").replace("myColumn", "My Column")
self.assertEqual(expected, label)