update

module5.ipynb

@@ -214,6 +214,42 @@
     "\n",
     "rearrange_name(\"Lovelace, Ada\")  "
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Writing unit tests in python"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#!/usr/bin/env python3\n",
+    "import re\n",
+    "def rearrange_name(name):\n",
+    "  result = re.search(r\"^([\\w .]*), ([\\w .]*)$\", name)\n",
+    "  return \"{} {}\".format(result[2], result[1])\n",
+    "\n",
+    "#!/usr/bin/env python3\n",
+    "import unittest\n",
+    "\n",
+    "from rearrange import rearrange_name\n",
+    "class TestRearrange(unittest.TestCase):\n",
+    "  def test_basic(self):\n",
+    "    testcase = \"Lovelace, Ada\"\n",
+    "    expected = \"Ada Lovelace\"\n",
+    "    self.assertEqual(rearrange_name(testcase), expected)\n",
+    "\n",
+    "# Run the tests\n",
+    "unittest.main()\n",
+    "\n",
+    "chmod +x rearrange_test.py \n",
+    "./rearrange_test.py "
+   ]
   }
  ],
  "metadata": {