code: def testVisualizationRegistrationAndUsage(self, *unused_mocks):
      registry = visualizations.ArtifactVisualizationRegistry()
      visualizations.get_registry = mock.MagicMock(
          return_value=registry
      )
      mock_object = mock.MagicMock()
  
      class MyVisualization(visualizations.ArtifactVisualization):
          # Arbitrary artifact type class.
          ARTIFACT_TYPE = standard_artifacts.Examples
  
          def display(self, unused_artifact):
              mock_object("foo")
  
      visualizations.get_registry().register(MyVisualization)
      self.assertIs(
          MyVisualization,
          visualizations.get_registry()
          .get_visualization(standard_artifacts.Examples.TYPE_NAME)
          .__class__,
      )
  
file_path: tfx/tfx/orchestration/experimental/interactive/visualizations_test.py
imports: import builtins
  from unittest import mock
  import tensorflow
  from tfx.orchestration.experimental.interactive import visualizations
  from tfx.types import standard_artifacts
label_desc: Use image node_id 3 for calling the VisualizationsTest obj's underlying member method code with example usage: obj.testVisualizationRegistrationAndUsage() without return types