2626from opentelemetry import trace
2727from opentelemetry .instrumentation .redis import RedisInstrumentor
2828from opentelemetry .semconv ._incubating .attributes .db_attributes import (
29- DB_SYSTEM ,
3029 DB_REDIS_DATABASE_INDEX ,
30+ DB_SYSTEM ,
3131 DbSystemValues ,
3232)
3333from opentelemetry .semconv ._incubating .attributes .net_attributes import (
3636 NET_TRANSPORT ,
3737 NetTransportValues ,
3838)
39-
4039from opentelemetry .test .test_base import TestBase
4140from opentelemetry .trace import SpanKind
4241
@@ -267,12 +266,8 @@ def test_attributes_default(self):
267266 span .attributes [DB_SYSTEM ],
268267 DbSystemValues .REDIS .value ,
269268 )
270- self .assertEqual (
271- span .attributes [DB_REDIS_DATABASE_INDEX ], 0
272- )
273- self .assertEqual (
274- span .attributes [NET_PEER_NAME ], "localhost"
275- )
269+ self .assertEqual (span .attributes [DB_REDIS_DATABASE_INDEX ], 0 )
270+ self .assertEqual (span .attributes [NET_PEER_NAME ], "localhost" )
276271 self .assertEqual (span .attributes [NET_PEER_PORT ], 6379 )
277272 self .assertEqual (
278273 span .attributes [NET_TRANSPORT ],
@@ -293,12 +288,8 @@ def test_attributes_tcp(self):
293288 span .attributes [DB_SYSTEM ],
294289 DbSystemValues .REDIS .value ,
295290 )
296- self .assertEqual (
297- span .attributes [DB_REDIS_DATABASE_INDEX ], 1
298- )
299- self .assertEqual (
300- span .attributes [NET_PEER_NAME ], "1.1.1.1"
301- )
291+ self .assertEqual (span .attributes [DB_REDIS_DATABASE_INDEX ], 1 )
292+ self .assertEqual (span .attributes [NET_PEER_NAME ], "1.1.1.1" )
302293 self .assertEqual (span .attributes [NET_PEER_PORT ], 6380 )
303294 self .assertEqual (
304295 span .attributes [NET_TRANSPORT ],
@@ -321,9 +312,7 @@ def test_attributes_unix_socket(self):
321312 span .attributes [DB_SYSTEM ],
322313 DbSystemValues .REDIS .value ,
323314 )
324- self .assertEqual (
325- span .attributes [DB_REDIS_DATABASE_INDEX ], 3
326- )
315+ self .assertEqual (span .attributes [DB_REDIS_DATABASE_INDEX ], 3 )
327316 self .assertEqual (
328317 span .attributes [NET_PEER_NAME ],
329318 "/path/to/socket.sock" ,
0 commit comments