def testCV2ImshowDisabled(self):
_cv2._register_hook()
cv2 = importlib.import_module("cv2")
self.assertNotIn("COLAB_CV2_IMPORT_HOOK_EXCEPTION", os.environ)
self.assertIn("cv2", sys.modules)
# Calling the function leads to the custom error.
with self.assertRaises(_cv2.DisabledFunctionError) as err:
cv2.imshow()
self.assertIn(_cv2._OpenCVImportHook.reason.format("cv2"),
str(err.exception))
# After Enabling, we get a TypeError (because we pass no arguments).
os.environ[_cv2._OpenCVImportHook.env_var] = "true"
with self.assertRaises(TypeError) as err:
cv2.imshow()
def _register_hooks(): _altair._register_hook() # pylint:disable=protected-access _bokeh._register_hook() # pylint:disable=protected-access _cv2._register_hook() # pylint:disable=protected-access