Fix #196 by calling numpy.random.seed().

plugin/PySrc/code_tracer.py

@@ -733,7 +733,7 @@ def find_module(self, fullname, path=None):
         if (fullname == self.module_name or
                 (fullname == SCOPE_NAME and self.is_own_driver)):
             return self
-        if fullname not in ('matplotlib', 'matplotlib.pyplot'):
+        if fullname not in ('matplotlib', 'matplotlib.pyplot', 'numpy.random'):
             return None
         is_after = False
         for finder in sys.meta_path:
@@ -742,11 +742,11 @@ def find_module(self, fullname, path=None):
                 continue
             loader = finder.find_module(fullname, path)
             if loader is not None:
-                return PatchedMatplotlibLoader(fullname, loader, self.is_zoomed)
+                return PatchedModuleLoader(fullname, loader, self.is_zoomed)
         if sys.version_info < (3, 0) and not TracedModuleImporter.is_desperate:
             # Didn't find anyone to load the module, get desperate.
             TracedModuleImporter.is_desperate = True
-            return PatchedMatplotlibLoader(fullname, None, self.is_zoomed)
+            return PatchedModuleLoader(fullname, None, self.is_zoomed)
 
     def load_module(self, fullname):
         if '.' in self.module_name:
@@ -768,7 +768,7 @@ def load_module(self, fullname):
         return new_mod
 
 
-class PatchedMatplotlibLoader(object):
+class PatchedModuleLoader(object):
     def __init__(self, fullname, main_loader, is_zoomed):
         self.fullname = fullname
         self.main_loader = main_loader
@@ -781,7 +781,9 @@ def load_module(self, fullname):
         else:
             module = import_module(fullname)
             TracedModuleImporter.is_desperate = False
-        if fullname == 'matplotlib':
+        if fullname == 'numpy.random':
+            module.seed(0)
+        elif fullname == 'matplotlib':
             module.use('Agg')
         elif fullname == 'matplotlib.pyplot':
             self.plt = module

test/PySrc/tests/test_code_tracer_matplotlib.py

@@ -20,6 +20,18 @@ def clear_matplotlib():
             yield True
 
 
+@pytest.fixture(name='is_numpy_cleared')
+def clear_numpy():
+    for should_yield in (True, False):
+        to_delete = [module_name
+                     for module_name in sys.modules
+                     if module_name.startswith('numpy')]
+        for module_name in to_delete:
+            del sys.modules[module_name]
+        if should_yield:
+            yield True
+
+
 def replace_image(report):
     report = trim_report(report)
     report = re.sub(r"image='[a-zA-Z0-9+/=]*'", "image='...'", report)
@@ -47,3 +59,22 @@ def test_show(is_matplotlib_cleared):
     report = tracer.trace_turtle(code)
 
     assert expected_report == replace_image(report)
+
+
+def test_numpy_random(is_numpy_cleared):
+    assert is_numpy_cleared
+    code = """\
+import numpy as np 
+
+x = np.random.normal(size=3)
+"""
+    expected_report = """\
+
+
+x = array([1.76405235, 0.40015721, 0.97873798])
+"""
+    tracer = CodeTracer()
+
+    report = tracer.trace_code(code)
+
+    assert expected_report == trim_report(report)