Skip to content

Commit

Permalink
remove individual ignores
Browse files Browse the repository at this point in the history
  • Loading branch information
sydney-runkle committed Oct 15, 2024
1 parent e1beed5 commit 377e7f1
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions tests/otel_integrations/test_psycopg.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
# can be removed once we upgrade to latest of psycopg-binary, see https://github.com/psycopg/psycopg/issues/708
# pyright: reportUnknownMemberType=false
# pyright: reportUnknownVariableType=false

import sys
from unittest import mock
Expand Down Expand Up @@ -30,12 +31,12 @@ def test_instrument_psycopg():

instrument_psycopg(psycopg)
assert original_connect is not psycopg.connect
PsycopgInstrumentor().uninstrument() # type: ignore
PsycopgInstrumentor().uninstrument()
assert original_connect is psycopg.connect

instrument_psycopg('psycopg')
assert original_connect is not psycopg.connect
PsycopgInstrumentor().uninstrument() # type: ignore
PsycopgInstrumentor().uninstrument()
assert original_connect is psycopg.connect


Expand All @@ -44,12 +45,12 @@ def test_instrument_psycopg2():

instrument_psycopg(psycopg2)
assert original_connect is not psycopg2.connect
Psycopg2Instrumentor().uninstrument() # type: ignore
Psycopg2Instrumentor().uninstrument()
assert original_connect is psycopg2.connect

instrument_psycopg('psycopg2')
assert original_connect is not psycopg2.connect
Psycopg2Instrumentor().uninstrument() # type: ignore
Psycopg2Instrumentor().uninstrument()
assert original_connect is psycopg2.connect


Expand All @@ -60,8 +61,8 @@ def test_instrument_both():
instrument_psycopg()
assert original_connect is not psycopg.connect
assert original_connect2 is not psycopg2.connect
PsycopgInstrumentor().uninstrument() # type: ignore
Psycopg2Instrumentor().uninstrument() # type: ignore
PsycopgInstrumentor().uninstrument()
Psycopg2Instrumentor().uninstrument()
assert original_connect is psycopg.connect
assert original_connect2 is psycopg2.connect

Expand Down

0 comments on commit 377e7f1

Please sign in to comment.