def METHOD_NAME(self, func_names, valid):
    base_viz = CustomBaseVisualization()
    batch = {func_name: "test" for func_name in func_names}
    if not valid:
        with pytest.raises(ValueError, match="Invalid function names"):
            base_viz.show(batch, RunningStage.TRAINING, func_names)
    else:
        base_viz.show(batch, RunningStage.TRAINING, func_names)
        for func_name in func_names:
            if hasattr(base_viz, f"show_{func_name}_called"):
                assert getattr(base_viz, f"show_{func_name}_called")