some unittest fixes

inducer · Apr 21, 2021 · 2578b3b · 2578b3b
1 parent 2b88d92
commit 2578b3b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/course/page/code_run_backend_octave.py b/course/page/code_run_backend_octave.py
@@ -201,6 +201,12 @@ def output_html(s):
     if getattr(run_req, "setup_code", None):
         try:
             oc.eval(run_req.setup_code)
+            # put variables from user in main context
+            for name in run_req.names_for_user:
+                try:
+                    maint_ctx[name] = oc.pull(name)
+                except oct2py.Oct2PyError:
+                    maint_ctx[name] = None
         except Exception:
             package_exception(result, "setup_error")
             return

diff --git a/tests/test_pages/markdowns.py b/tests/test_pages/markdowns.py
@@ -469,6 +469,7 @@
     # Adding two numbers in Octave
 
 setup_code: |
+    pkg load statistics
     a = unifrnd(-10,10)
     b = unifrnd(-10,10)
 
@@ -507,6 +508,7 @@
     # Adding two numbers in Octave
 
 setup_code: |
+    pkg load statistics
     a = unifrnd(-10,10)
     b = unifrnd(-10,10)
 
@@ -544,6 +546,7 @@
     # Adding two numbers in Octave
 
 setup_code: |
+    pkg load statistics
     a = unifrnd(-10,10)
     b = unifrnd(-10,10)
 
@@ -581,6 +584,7 @@
     # Adding two numbers in Octave
 
 setup_code: |
+    pkg load statistics
     a = unifrnd(-10,10)
     b = unifrnd(-10,10)
 
@@ -616,6 +620,7 @@
     # Adding two numbers in Octave
 
 setup_code: |
+    pkg load statistics
     a = unifrnd(-10,10)
     b = unifrnd(-10,10)
 
@@ -650,6 +655,7 @@
     # Adding two numbers in Octave
 
 setup_code: |
+    pkg load statistics
     a = unifrnd(-10,10)
     b = unifrnd(-10,10)
 
@@ -672,6 +678,7 @@
     # Adding two numbers in Octave
 
 setup_code: |
+    pkg load statistics
     a = unifrnd(-10,10)
     b = unifrnd(-10,10)
 
@@ -703,6 +710,7 @@
     # Adding two numbers in Octave
 
 setup_code: |
+    pkg load statistics
     a = unifrnd(-10,10)
     b = unifrnd(-10,10)
 
@@ -737,6 +745,7 @@
     # Adding two numbers in Octave
 
 setup_code: |
+    pkg load statistics
     a = unifrnd(-10,10)
     b = unifrnd(-10,10)