Extract duplicated helper code from the notebooks spawning python interpreters (#307)

Author: eric-wieser

examples/ipython/LaTeX.ipynb

@@ -6,28 +6,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import subprocess\n",
-    "import sys\n",
     "import os\n",
-    "from IPython.display import display_pretty\n",
-    "\n",
-    "os.chdir('../LaTeX')\n",
-    "\n",
-    "def check(name):\n",
-    "    # todo: use subprocess.run once we drop Python 2.7\n",
-    "    p = subprocess.Popen([sys.executable, name + '.py'], stderr=subprocess.PIPE, universal_newlines=True)\n",
-    "    try:\n",
-    "        stdout, stderr = p.communicate()\n",
-    "    except:\n",
-    "        p.kill()\n",
-    "    if p.poll():\n",
-    "        raise RuntimeError(\"The script raised an exception:\\n\\n\" + stderr)\n",
-    "\n",
-    "    with open(name + '.tex', 'r') as f:\n",
-    "        # can't use display.Latex here, it would result in CSS comparisons in the output.\n",
-    "        \n",
-    "        # using `display` forces this to be a separate output to any stdout from above.\n",
-    "        display_pretty(f.read(), raw=True)"
+    "from galgebra_ipython_helpers import check\n",
+    "os.chdir('../LaTeX')"
    ]
   },
   {

examples/ipython/Old Format.ipynb

@@ -6,28 +6,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import subprocess\n",
-    "import sys\n",
     "import os\n",
-    "from IPython.display import display_pretty\n",
-    "\n",
-    "os.chdir('../Old Format')\n",
-    "\n",
-    "def check_latex(name):\n",
-    "    # todo: use subprocess.run once we drop Python 2.7\n",
-    "    p = subprocess.Popen([sys.executable, name + '.py'], stderr=subprocess.PIPE, universal_newlines=True)\n",
-    "    try:\n",
-    "        stdout, stderr = p.communicate()\n",
-    "    except:\n",
-    "        p.kill()\n",
-    "    if p.poll():\n",
-    "        raise RuntimeError(\"The script raised an exception:\\n\\n\" + stderr)\n",
-    "\n",
-    "    with open(name + '.tex', 'r') as f:\n",
-    "        # can't use display.Latex here, it would result in CSS comparisons in the output.\n",
-    "        \n",
-    "        # using `display` forces this to be a separate output to any stdout from above.\n",
-    "        display_pretty(f.read(), raw=True)"
+    "from galgebra_ipython_helpers import check as check_latex, run\n",
+    "os.chdir('../Old Format')"
    ]
   },
   {
@@ -50,7 +31,7 @@
     }
    ],
    "source": [
-    "!python bad_example.py"
+    "run('bad_example')"
    ]
   },
   {
@@ -86,7 +67,7 @@
     }
    ],
    "source": [
-    "!python eval_check.py"
+    "run('eval_check')"
    ]
   },
   {
@@ -106,7 +87,7 @@
     }
    ],
    "source": [
-    "!python exp_check.py"
+    "run('exp_check')"
    ]
   },
   {
@@ -618,7 +599,7 @@
     }
    ],
    "source": [
-    "!python mv_setup_options.py"
+    "run('mv_setup_options')"
    ]
   },
   {
@@ -925,7 +906,7 @@
     }
    ],
    "source": [
-    "!python prob_not_solenoidal.py"
+    "run('prob_not_solenoidal')"
    ]
   },
   {
@@ -1113,7 +1094,7 @@
     }
    ],
    "source": [
-    "!python products_latex.py"
+    "run('products_latex')"
    ]
   },
   {
@@ -1295,7 +1276,7 @@
     }
    ],
    "source": [
-    "!python simple_check.py"
+    "run('simple_check')"
    ]
   },
   {
@@ -1548,7 +1529,7 @@
     }
    ],
    "source": [
-    "!python terminal_check.py"
+    "run('terminal_check')"
    ]
   }
  ],

examples/ipython/galgebra_ipython_helpers.py

@@ -0,0 +1,24 @@
+import subprocess
+import sys
+from IPython.display import display_pretty
+
+
+def check(name):
+    # todo: use subprocess.run once we drop Python 2.7
+    p = subprocess.Popen([sys.executable, name + '.py'], stderr=subprocess.PIPE, universal_newlines=True)
+    try:
+        stdout, stderr = p.communicate()
+    except:
+        p.kill()
+    if p.poll():
+        raise RuntimeError("The script raised an exception:\n\n" + stderr)
+
+    with open(name + '.tex', 'r') as f:
+        # can't use display.Latex here, it would result in CSS comparisons in the output.
+
+        # using `display` forces this to be a separate output to any stdout from above.
+        display_pretty(f.read(), raw=True)
+
+
+def run(name):
+    get_ipython().system('python {}.py'.format(name))