Fix tests

open-telemetry · Jun 3, 2021 · 97cfe11 · 97cfe11
1 parent e3e880e
commit 97cfe11
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/util/src/opentelemetry/test/__init__.py b/tests/util/src/opentelemetry/test/__init__.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 
-from traceback import extract_tb
+from traceback import format_tb
 
 
 class AssertNotRaisesMixin:
@@ -27,9 +27,6 @@ def __enter__(self):
         def __exit__(self, type_, value, tb):  # pylint: disable=invalid-name
             if value is not None and type_ in self._exception_types:
 
-                from ipdb import set_trace
-                set_trace()
-
                 self._test_case.fail(
                     "Unexpected exception was raised:\n{}".format(
                         "\n".join(format_tb(tb))
@@ -47,3 +44,4 @@ def __init__(self, *args, **kwargs):
 
         super().__init__(*args, **kwargs)
         # pylint: disable=invalid-name
+        self.assertNotRaises = self._AssertNotRaises(self)