doc: mocking artiq

m-labs · Jul 19, 2017 · b15c76c · b15c76c
1 parent fdb27e9
commit b15c76c
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 3 deletions.
diff --git a/doc/conf.py b/doc/conf.py
@@ -27,7 +27,19 @@ def __getattr__(name):
         return Mock()
 
 
-mock_modules = ["numpy", "scipy", "scipy.interpolate", "serial"]
+mock_modules = [
+    "numpy",
+    "scipy",
+    "scipy.interpolate",
+    "serial",
+    "artiq",
+    "artiq.protocols",
+    "artiq.tools",
+    "artiq.language",
+    "artiq.language.core",
+    "artiq.coredevice",
+    "artiq.protocols.pc_rpc",
+]
 
 for module in mock_modules:
     sys.modules[module] = Mock()

diff --git a/pdq/artiq/aqctl_pdq.py b/pdq/artiq/aqctl_pdq.py
@@ -6,7 +6,8 @@
 
 from pdq.host.usb import PDQ
 from artiq.protocols.pc_rpc import simple_server_loop
-from artiq.tools import *
+from artiq.tools import (verbosity_args, simple_network_args, init_logger,
+    bind_address_from_args)
 
 
 def get_argparser():

diff --git a/pdq/artiq/mediator.py b/pdq/artiq/mediator.py
@@ -1,4 +1,4 @@
-from artiq.language import *
+from artiq.language import us, ns, delay_mu, at_mu, kernel, portable, s
 
 
 frame_setup = 1.5*us