move autograder helper code into its own file

Author: afeld

extras/__init__.py

@@ -0,0 +1,29 @@
+import ast
+import os
+import pytest
+from .nb_helper import *
+
+
+class InputChecker(ast.NodeVisitor):
+    def __init__(self):
+        self.num_inputs = 0
+
+    def visit_Call(self, node):
+        if isinstance(node.func, ast.Name) and node.func.id == "input":
+            self.num_inputs += 1
+
+
+@pytest.fixture()
+def notebook():
+    nb_full_path = os.path.join(os.getcwd(), "hw_0.ipynb")
+    return read_notebook(nb_full_path)
+
+
+@pytest.fixture()
+def script(notebook):
+    return notebook_to_script(notebook)
+
+
+@pytest.fixture()
+def tree(script):
+    return ast.parse(script)

extras/scripts/hw_0_helper.py

@@ -196,58 +196,18 @@
   {
    "cell_type": "code",
    "execution_count": 1,
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "application/javascript": [
-       "IPython.notebook.save_checkpoint();\n",
-       "IPython.notebook.kernel.execute('nb_name = \"' + IPython.notebook.notebook_name + '\"')\n"
-      ],
-      "text/plain": [
-       "<IPython.core.display.Javascript object>"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
-   "source": [
-    "%%javascript\n",
-    "IPython.notebook.save_checkpoint();\n",
-    "IPython.notebook.kernel.execute('nb_name = \"' + IPython.notebook.notebook_name + '\"')"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
    "metadata": {
     "scrolled": true
    },
    "outputs": [],
    "source": [
-    "import ast\n",
     "import ipytest\n",
-    "import nbconvert\n",
-    "import nbformat\n",
-    "import os\n",
-    "\n",
-    "\n",
-    "ipytest.autoconfig()\n",
-    "\n",
-    "\n",
-    "class InputChecker(ast.NodeVisitor):\n",
-    "    def __init__(self):\n",
-    "        self.num_inputs = 0\n",
-    "\n",
-    "    def visit_Call(self, node):\n",
-    "        if isinstance(node.func, ast.Name) and node.func.id == \"input\":\n",
-    "            self.num_inputs += 1"
+    "ipytest.autoconfig()"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [
     {
@@ -258,7 +218,7 @@
       "============================================= FAILURES =============================================\n",
       "\u001b[31m\u001b[1m_________________________________________ test_num_inputs __________________________________________\u001b[0m\n",
       "\n",
-      "tree = <ast.Module object at 0x10d086dd0>\n",
+      "tree = <ast.Module object at 0x109a5a860>\n",
       "\n",
       "    \u001b[94mdef\u001b[39;49;00m \u001b[92mtest_num_inputs\u001b[39;49;00m(tree):\n",
       "        checker = InputChecker()\n",
@@ -267,43 +227,24 @@
       ">       \u001b[94massert\u001b[39;49;00m checker.num_inputs >= \u001b[94m8\u001b[39;49;00m, \u001b[33m\"\u001b[39;49;00m\u001b[33myou don\u001b[39;49;00m\u001b[33m'\u001b[39;49;00m\u001b[33mt have enough calls to input()\u001b[39;49;00m\u001b[33m\"\u001b[39;49;00m\n",
       "\u001b[1m\u001b[31mE       AssertionError: you don't have enough calls to input()\u001b[0m\n",
       "\u001b[1m\u001b[31mE       assert 2 >= 8\u001b[0m\n",
-      "\u001b[1m\u001b[31mE        +  where 2 = <__main__.InputChecker object at 0x10d084fa0>.num_inputs\u001b[0m\n",
+      "\u001b[1m\u001b[31mE        +  where 2 = <extras.scripts.hw_0_helper.InputChecker object at 0x10acc9ba0>.num_inputs\u001b[0m\n",
       "\n",
-      "\u001b[1m\u001b[31m/var/folders/kg/1ys0dccx4237f5wsd_w10dt80000gn/T/ipykernel_65587/682324548.py\u001b[0m:27: AssertionError\n",
+      "\u001b[1m\u001b[31m/var/folders/kg/1ys0dccx4237f5wsd_w10dt80000gn/T/ipykernel_67029/2901258849.py\u001b[0m:8: AssertionError\n",
       "===================================== short test summary info ======================================\n",
-      "FAILED tmpciht1t3f.py::test_num_inputs - AssertionError: you don't have enough calls to input()\n"
+      "FAILED tmpyoyk9z3m.py::test_num_inputs - AssertionError: you don't have enough calls to input()\n"
      ]
     }
    ],
    "source": [
     "%%ipytest -qq\n",
     "\n",
-    "import pytest\n",
-    "\n",
-    "\n",
-    "\n",
-    "@pytest.fixture()\n",
-    "def notebook():\n",
-    "    nb_full_path = os.path.join(os.getcwd(), nb_name)\n",
-    "    return nbformat.read(nb_full_path, as_version=4)\n",
-    "\n",
-    "\n",
-    "@pytest.fixture()\n",
-    "def script(notebook):\n",
-    "    exporter = nbconvert.exporters.PythonExporter\n",
-    "    script, _ = nbconvert.exporters.export(exporter, notebook)\n",
-    "    return script\n",
-    "\n",
-    "\n",
-    "@pytest.fixture()\n",
-    "def tree(script):\n",
-    "    return ast.parse(script)\n",
+    "from extras.scripts.hw_0_helper import *\n",
     "\n",
     "\n",
     "def test_num_inputs(tree):\n",
     "    checker = InputChecker()\n",
     "    checker.visit(tree)\n",
-    "    \n",
+    "\n",
     "    assert checker.num_inputs >= 8, \"you don't have enough calls to input()\""
    ]
   },