fix: Remove the proper attribute after testing

tektronix · Sep 17, 2024 · fba127d · fba127d
1 parent dc022c4
commit fba127d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_extension_mixin.py b/tests/test_extension_mixin.py
@@ -100,7 +100,7 @@ def _remove_added_methods() -> Iterator[None]:
         (AFG3K, "custom_model_getter_afg3k"),
         (AFG3KC, "custom_model_getter_afg3kc"),
         (PIDevice, "added_method"),
-        (TSPDevice, "added_method"),
+        (TSPDevice, "added_tsp_method"),
     ):
         with contextlib.suppress(AttributeError):
             delattr(obj, name)