From b877657456d977912052ac0cdc926750a67ba89d Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 3 Nov 2021 15:01:23 +0800
Subject: [PATCH] ci(pytest): fixture doctest namespace in pytest config

GitOrigin-RevId: 90ef1924dc69311f0945aa3d03923af968439c6f
---
 imperative/python/megengine/conftest.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 imperative/python/megengine/conftest.py

diff --git a/imperative/python/megengine/conftest.py b/imperative/python/megengine/conftest.py
new file mode 100644
index 00000000..4a20f181
--- /dev/null
+++ b/imperative/python/megengine/conftest.py
@@ -0,0 +1,21 @@
+import pytest
+
+import megengine
+
+
+@pytest.fixture(autouse=True)
+def import_megengine_path(doctest_namespace):
+    doctest_namespace["mge"] = megengine
+    doctest_namespace["Tensor"] = megengine.Tensor
+    doctest_namespace["F"] = megengine.functional
+    doctest_namespace["M"] = megengine.module
+    doctest_namespace["Q"] = megengine.quantization
+    doctest_namespace["data"] = megengine.data
+    doctest_namespace["autodiff"] = megengine.autodiff
+    doctest_namespace["optim"] = megengine.optimizer
+    doctest_namespace["jit"] = megengine.jit
+    doctest_namespace["amp"] = megengine.amp
+    doctest_namespace["dist"] = megengine.distributed
+    doctest_namespace["tm"] = megengine.traced_module
+    doctest_namespace["hub"] = megengine.hub
+    doctest_namespace["utils"] = megengine.utils