def test_custom_default_name(self, small_cont: ImageContainer):
custom_features = small_cont.features_custom(np.mean, layer="image", channels=[0])
summary_features = small_cont.features_summary("image", feature_name="summary", channels=[0])
assert len(custom_features) == 1
assert f"{np.mean.__name__}_0" in custom_features
assert custom_features[f"{np.mean.__name__}_0"] == summary_features["summary_ch-0_mean"]
def test_custom_returns_iterable(self, small_cont: ImageContainer):
def dummy(_: np.ndarray) -> Tuple[int, int]:
return 0, 1
features = small_cont.features_custom(dummy, layer="image", feature_name="foo")
assert len(features) == 2
assert features["foo_0"] == 0
assert features["foo_1"] == 1
