opensource/opentelemetry-python-contrib
1
0
Fork 0
mirror of https://github.com/open-telemetry/opentelemetry-python-contrib.git synced 2025-08-06 14:59:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

missing test

Browse Source
This commit is contained in:
flynn
2020-11-05 21:40:13 -05:00
parent e420c184b4
commit 369dba4480
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
instrumentation/opentelemetry-instrumentation-sklearn/tests/test_sklearn.py
View File

@ -159,3 +159,17 @@ class TestSklearn(TestBase):
model.predict(x_test) model.predict(x_test)
spans = self.memory_exporter.get_finished_spans() spans = self.memory_exporter.get_finished_spans()
self.assertEqual(len(spans), 0) 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"