mock some moduless

NetEaseGame · Mar 13, 2016 · 78e329a · 78e329a
1 parent 1e535bd
commit 78e329a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/atx/device.py b/atx/device.py
@@ -17,14 +17,9 @@
 import warnings
 import logging
 
-try:
-    import xkjxxx
-    import cv2
-    import numpy as np
-    import aircv as ac
-except:
-    warnings.warn("OpenCV and Numpy need to be installed.", ImportWarning)
-
+import cv2
+import numpy as np
+import aircv as ac
 from uiautomator import device as d
 from uiautomator import Device as UiaDevice
 from uiautomator import AutomatorDeviceObject

diff --git a/docs/conf.py b/docs/conf.py
@@ -293,3 +293,12 @@
 
 # If true, do not generate a @detailmenu in the "Top" node's menu.
 #texinfo_no_detailmenu = False
+
+# mock some functions
+class Mock(object):
+    @classmethod
+    def __getattr__(cls, name):
+            return Mock()
+
+MOCK_MODULES = ['cv2', 'numpy']
+sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)