def test_maybe_numeric_slice(self):
df = DataFrame({"A": [1, 2], "B": ["c", "d"], "C": [True, False]})
result = maybe_numeric_slice(df, slice_=None)
expected = pd.IndexSlice[:, ["A"]]
assert result == expected
result = maybe_numeric_slice(df, None, include_bool=True)
expected = pd.IndexSlice[:, ["A", "C"]]
result = maybe_numeric_slice(df, [1])
expected = [1]
assert result == expected
def get_heatmap(self, cmap, sub=None, text_color_threshold=0.408):
if sub is None:
sub = self.subset
cmap = self.min_max_scaler(sub).applymap(
lambda x: self.get_heatmap_color(x, cmap, text_color_threshold))
sub = maybe_numeric_slice(self.data, sub)
sub = non_reducing_slice(sub)
c_mapping = lambda x: cmap.loc[sub]
self.apply(c_mapping, subset=sub, axis=None)
return self
