diff --git a/tests/trace_service.py b/tests/trace_service.py index 291a984d..47e477db 100644 --- a/tests/trace_service.py +++ b/tests/trace_service.py @@ -23,7 +23,10 @@ def get_batch(self): return self.batches.pop() def get_span(self): - return self.get_batch()[0].scope_spans[0].spans.pop() + batch = self.get_batch() + assert len(batch.scope_spans) == 1 + assert len(batch.scope_spans[0].spans) == 1 + return batch.scope_spans[0].spans.pop() @pytest.fixture(scope="module")