nose experiment.ipynb

{
 "metadata": {
  "name": "nose experiment"
 },
 "nbformat": 3,
 "nbformat_minor": 0,
 "worksheets": [
  {
   "cells": [
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "import sys, os\n",
      "import logging\n",
      "import unittest\n",
      "\n",
      "log = logging.getLogger()\n",
      "\n",
      "from nose import core, loader\n",
      "\n",
      "#logging.basicConfig(level=logging.DEBUG)\n",
      "\n",
      "from types import ModuleType"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [],
     "prompt_number": 1
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "def adder(a, b):\n",
      "    return a + b\n",
      "\n",
      "def test_nothing():\n",
      "    assert 1+1 == 2\n",
      "\n",
      "def test_adder():\n",
      "    assert adder(3, 4) == 7\n",
      "    assert adder(-3, -5) == -8\n",
      "    \n",
      "import random\n",
      "def test_random_generated_tests():\n",
      "    for _ in range(random.randint(1, 100)):\n",
      "        yield lambda: None\n",
      "\n",
      "flake = False\n",
      "def test_flake():\n",
      "    global flake\n",
      "    flake = not flake\n",
      "    assert flake"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [],
     "prompt_number": 2
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "class MyProgram(core.TestProgram):\n",
      "    # XXX yuck: copy superclass runTests() so we can instantiate our own runner class;\n",
      "    # can't do it early because we don't have access to nose's config object.\n",
      "    def runTests(self):\n",
      "        self.testRunner = MyRunner(self.config)\n",
      "        # the rest is mostly duplicate code ;-(\n",
      "        plug_runner = self.config.plugins.prepareTestRunner(self.testRunner)\n",
      "        if plug_runner is not None:\n",
      "            self.testRunner = plug_runner\n",
      "        self.result = self.testRunner.run(self.test)\n",
      "        self.success = self.result.wasSuccessful()\n",
      "        return self.success\n",
      "\n",
      "class MyResult(unittest.TestResult):\n",
      "    def make_bar(self, tests, failing):\n",
      "        return '''<div>\n",
      "                <div style=\"background:red; width:%dpx\">&nbsp;</div>\n",
      "                <div style=\"background:green; width:%dpx\">&nbsp;</div>\n",
      "                </div>''' % (\n",
      "            failing * 10, (tests - failing) * 10)\n",
      "\n",
      "    def make_table_of_tests(self, tests):\n",
      "        table = '<table>'\n",
      "        for test in tests:\n",
      "            table += '<tr><td>%s</td><td><pre><![CDATA[%s]]></pre></td>' % (\n",
      "                str(test[0]), test[1])\n",
      "        table += '</table>'\n",
      "        return table\n",
      "    \n",
      "    def _repr_html_(self):\n",
      "        if self.errors or self.failures:\n",
      "            not_successes = len(self.errors) + len(self.failures)\n",
      "            return self.make_bar(self.testsRun, not_successes) + \\\n",
      "                '''\n",
      "                <h2 style=\"color:red\">Errors</h2>%s\n",
      "                <h2 style=\"color:red\">Failures</h2>%s''' % (\n",
      "                    self.make_table_of_tests(self.errors),\n",
      "                    self.make_table_of_tests(self.failures))\n",
      "        else:\n",
      "            return self.make_bar(self.testsRun, 0) + '<div>%d/%d&nbsp;tests&nbsp;passed</div>' % (\n",
      "                self.testsRun, self.testsRun)\n",
      "\n",
      "class MyRunner(object):\n",
      "    def __init__(self, config):\n",
      "        self.config = config\n",
      "\n",
      "    def run(self, test):\n",
      "        result = MyResult()\n",
      "        if hasattr(result, 'startTestRun'):   # python 2.7\n",
      "            result.startTestRun()\n",
      "        test(result)\n",
      "        if hasattr(result, 'stopTestRun'):\n",
      "            result.stopTestRun()\n",
      "        self.config.plugins.finalize(result)\n",
      "        self.result = result\n",
      "        return result"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [],
     "prompt_number": 3
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "test_module = ModuleType('test_module')\n",
      "test_module.__dict__.update(get_ipython().user_ns)\n",
      "\n",
      "ldr = loader.TestLoader()\n",
      "tests = ldr.loadTestsFromModule(test_module)\n",
      "#print(\"discovered: %r\" % list(tests._tests))\n",
      "\n",
      "tprog = MyProgram(argv=['dummy'], exit=False, suite=tests)\n",
      "tprog.result"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [
      {
       "html": [
        "<div>\n",
        "                <div style=\"background:red; width:0px\">&nbsp;</div>\n",
        "                <div style=\"background:green; width:620px\">&nbsp;</div>\n",
        "                </div><div>62/62&nbsp;tests&nbsp;passed</div>"
       ],
       "output_type": "pyout",
       "prompt_number": 4,
       "text": [
        "<__main__.MyResult run=62 errors=0 failures=0>"
       ]
      }
     ],
     "prompt_number": 4
    }
   ],
   "metadata": {}
  }
 ]
}