code: def test_brute_force_topk_sampler_success(self):
      self.assertProtoEquals(
          """
        similarity_type: COSINE
      """,
          cs_config_builder.brute_force_topk_sampler("COSINE"),
      )
      self.assertProtoEquals(
          """
        similarity_type: COSINE
      """,
          cs_config_builder.brute_force_topk_sampler(
              cs_config_pb2.COSINE
          ),
      )
      self.assertProtoEquals(
          """
        similarity_type: DOT_PRODUCT
      """,
          cs_config_builder.brute_force_topk_sampler("DOT_PRODUCT"),
      )
      self.assertProtoEquals(
          """
        similarity_type: DOT_PRODUCT
      """,
          cs_config_builder.brute_force_topk_sampler(
              cs_config_pb2.DOT_PRODUCT
          ),
      )
  
file_path: neural-structured-learning/research/carls/candidate_sampling/candidate_sampler_config_builder_test.py
imports: from research.carls.candidate_sampling import candidate_sampler_config_builder
  from research.carls.candidate_sampling import candidate_sampler_config_pb2
  import tensorflow
label_desc: Use image node_id 2 for calling the CandidateSamplerConfigBuilderTest obj's underlying member method code with example usage: obj.test_brute_force_topk_sampler_success() without return types