Skip to content

CLN: Todo test reductions in test_style.py #44750

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Dec 4, 2021
Merged

CLN: Todo test reductions in test_style.py #44750

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
b6252f3
mi names
attack68 Dec 4, 2021
4f980f2
mi names
attack68 Dec 4, 2021
94061c1
remove duplicated test
attack68 Dec 4, 2021
d1863f0
add visibility check
attack68 Dec 4, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
117 changes: 24 additions & 93 deletions pandas/tests/io/formats/style/test_style.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -574,36 +574,6 @@ def test_index_name(self):
}
assert expected.items() <= result["head"][1][0].items()

def test_multiindex_name(self):
# https://github.com/pandas-dev/pandas/issues/11655
df = DataFrame({"A": [1, 2], "B": [3, 4], "C": [5, 6]})
result = df.set_index(["A", "B"]).style._translate(True, True)

expected = [
{
"class": "index_name level0",
"type": "th",
"value": "A",
"is_visible": True,
"display_value": "A",
},
{
"class": "index_name level1",
"type": "th",
"value": "B",
"is_visible": True,
"display_value": "B",
},
{
"class": "blank col0",
"type": "th",
"value": self.blank_value,
"is_visible": True,
"display_value": self.blank_value,
},
]
assert result["head"][1] == expected

def test_numeric_columns(self):
# https://github.com/pandas-dev/pandas/issues/12125
# smoke test for _translate
Expand Down Expand Up @@ -1037,7 +1007,7 @@ def test_get_level_lengths_un_sorted(self):
tm.assert_dict_equal(result, expected)

def test_mi_sparse_index_names(self):
# TODO this test is verbose can be minimised to more directly target test
# Test the class names and displayed value are correct on rendering MI names
df = DataFrame(
{"A": [1, 2]},
index=MultiIndex.from_arrays(
Expand All @@ -1049,28 +1019,22 @@ def test_mi_sparse_index_names(self):
expected = [
{
"class": "index_name level0",
"value": "idx_level_0",
"type": "th",
"is_visible": True,
"display_value": "idx_level_0",
"is_visible": True,
},
{
"class": "index_name level1",
"value": "idx_level_1",
"type": "th",
"is_visible": True,
"display_value": "idx_level_1",
"is_visible": True,
},
{
"class": "blank col0",
"value": self.blank_value,
"type": "th",
"is_visible": True,
"display_value": self.blank_value,
"is_visible": True,
},
]

assert head == expected
for i, expected_dict in enumerate(expected):
assert expected_dict.items() <= head[i].items()

def test_mi_sparse_column_names(self):
df = DataFrame(
Expand All @@ -1080,60 +1044,27 @@ def test_mi_sparse_column_names(self):
names=["idx_level_0", "idx_level_1"],
),
columns=MultiIndex.from_arrays(
[["C1", "C1", "C2", "C2"], [1, 0, 1, 0]], names=["col_0", "col_1"]
[["C1", "C1", "C2", "C2"], [1, 0, 1, 0]], names=["colnam_0", "colnam_1"]
),
)
result = Styler(df, cell_ids=False)._translate(True, True)
head = result["head"][1]
expected = [
{
"class": "blank",
"value": self.blank_value,
"display_value": self.blank_value,
"type": "th",
"is_visible": True,
},
{
"class": "index_name level1",
"value": "col_1",
"display_value": "col_1",
"is_visible": True,
"type": "th",
},
{
"class": "col_heading level1 col0",
"display_value": "1",
"is_visible": True,
"type": "th",
"value": 1,
"attributes": "",
},
{
"class": "col_heading level1 col1",
"display_value": "0",
"is_visible": True,
"type": "th",
"value": 0,
"attributes": "",
},
{
"class": "col_heading level1 col2",
"display_value": "1",
"is_visible": True,
"type": "th",
"value": 1,
"attributes": "",
},
{
"class": "col_heading level1 col3",
"display_value": "0",
"is_visible": True,
"type": "th",
"value": 0,
"attributes": "",
},
]
assert head == expected

for level in [0, 1]:
head = result["head"][level]
expected = [
{
"class": "blank",
"display_value": self.blank_value,
"is_visible": True,
},
{
"class": f"index_name level{level}",
"display_value": f"colnam_{level}",
"is_visible": True,
},
]
for i, expected_dict in enumerate(expected):
assert expected_dict.items() <= head[i].items()

def test_hide_column_headers(self):
ctx = self.styler.hide(axis="columns")._translate(True, True)
Expand Down