def METHOD_NAME(self):
    model = pipeline()
    attributes = {"model_name": "random_forest_model"}
    ski = SklearnInstrumentor()
    ski.instrument_estimator(estimator=model, attributes=attributes)
    x_test = random_input()
    model.predict(x_test)
    spans = self.memory_exporter.get_finished_spans()
    for span in spans:
        assert span.attributes["model_name"] == "random_forest_model"