def test_display(self):
report = Report(
self.data,
self.cols,
model_name=self.model_name,
dataset_name=self.dataset_name,
)
report.sort()
figure = report.figure()
figure.show()
report.sort(category_order={"Cat C": 1, "Cat A": 2, "Cat B": 3})
report.rename(slice_map={"Slice A": "A"}, category_map={"Cat B": "B"})
report.filter(slices=["A", "Slice B", "Slice C"])
report.set_range("f1", 0.05, 0.45)
report.update_config(font_size_heading=16)
figure = report.figure(show_title=True)
figure.show()
def test_rename(self):
report = Report(
self.data,
self.cols,
model_name=self.model_name,
dataset_name=self.dataset_name,
)
category_map = {"Cat C": "Cat D"}
slice_map = {"Slice A": "Slice D"}
report.rename(category_map=category_map, slice_map=slice_map)
actual = report.data
expected = pd.DataFrame([
["Cat A", "Slice C", 0.1, 5, [0.1, 0.2, 0.7], 300],
["Cat D", "Slice D", 0.2, 10, [0.4, 0.2, 0.4], 3],
["Cat A", "Slice D", 0.3, 15, [0.1, 0, 0.9], 5000],
["Cat B", "Slice B", 0.4, 20, [0.5, 0.4, 0.1], 812],
["Cat B", "Slice D", 0.5, 25, [0.3, 0.2, 0.5], 13312],
])
self.assertTrue(actual.equals(expected))