def METHOD_NAME(self) -> None:
    input_data = torch.rand(2, 2)
    transformed_data = torch.rand(1, 1)
    def mock_transform(data):
        return transformed_data
    feature = TextFeature(
        name=self.FEATURE_NAME,
        baseline_transforms=None,
        input_transforms=None,
        visualization_transform=mock_transform,
    )
    feature_output = feature.visualize(
        attribution=torch.rand(1, 1), data=input_data, contribution_frac=1.0
    )
    # has transformed data
    self.assertEqual(feature_output.base, transformed_data)
    feature = TextFeature(
        name=self.FEATURE_NAME,
        baseline_transforms=None,
        input_transforms=None,
        visualization_transform=None,
    )
    feature_output = feature.visualize(
        attribution=torch.rand(1, 1), data=input_data, contribution_frac=1.0
    )
    # has original data
    self.assertIs(feature_output.base, input_data)