From 369dba4480cbd2a042acab8478f2b9a50eedaf77 Mon Sep 17 00:00:00 2001 From: flynn Date: Thu, 5 Nov 2020 21:40:13 -0500 Subject: [PATCH] missing test --- .../tests/test_sklearn.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/instrumentation/opentelemetry-instrumentation-sklearn/tests/test_sklearn.py b/instrumentation/opentelemetry-instrumentation-sklearn/tests/test_sklearn.py index 2aa43e700..df454b6fa 100644 --- a/instrumentation/opentelemetry-instrumentation-sklearn/tests/test_sklearn.py +++ b/instrumentation/opentelemetry-instrumentation-sklearn/tests/test_sklearn.py @@ -159,3 +159,17 @@ class TestSklearn(TestBase): model.predict(x_test) spans = self.memory_exporter.get_finished_spans() self.assertEqual(len(spans), 0) + + def test_span_attributes(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"