Test preserving assignment not recognised as parameter

Author: takluyver

tests/sample.ipynb

@@ -28,6 +28,7 @@
     "print(\"This should be ignored\")\n",
     "\n",
     "café = \"καφές\"  # Not only ASCII\n",
+    "other_assignment = b ** 2  # Not recognised as a parameter\n",
     "\n",
     "def func():\n",
     "    not_a_parameter = 5  # Should not be found\n",

tests/test_basic.py

@@ -62,8 +62,9 @@ def test_rebuild(self):
         assert ns['d'] == True
         assert ns['café'] == "καφές"
 
-        # Function should be preserved
+        # Function & non parameter assignment should be preserved
         assert ns['func']() == 0.25
+        assert ns['other_assignment'] == 21 ** 2
 
     def test_rebuild_from_dict(self):
         new_params = self.param_dict.copy()
@@ -84,8 +85,9 @@ def test_rebuild_from_dict(self):
         assert ns['café'] == "καφές"
         assert ns['thé'] == "चाय"
 
-        # Function should be preserved
+        # Function & non parameter assignment should be preserved
         assert ns['func']() == 0.75
+        assert ns['other_assignment'] == 12 ** 2
 
     def test_new_values(self):
         params = code.parameter_values(self.params,