| 1 | #!/usr/bin/env python |
| 2 | |
| 3 | __author__ = "Patrick K. O'Brien <pobrien@orbtech.com>" |
| 4 | __cvsid__ = "$Id$" |
| 5 | __revision__ = "$Revision$"[11:-2] |
| 6 | |
| 7 | import unittest |
| 8 | |
| 9 | # Import from this module's parent directory. |
| 10 | import os |
| 11 | import sys |
| 12 | sys.path.insert(0, os.pardir) |
| 13 | import interpreter |
| 14 | del sys.path[0] |
| 15 | del sys |
| 16 | del os |
| 17 | |
| 18 | |
| 19 | """ |
| 20 | These unittest methods are preferred: |
| 21 | ------------------------------------- |
| 22 | self.assert_(expr, msg=None) |
| 23 | self.assertEqual(first, second, msg=None) |
| 24 | self.assertRaises(excClass, callableObj, *args, **kwargs) |
| 25 | self.fail(msg=None) |
| 26 | self.failIf(expr, msg=None) |
| 27 | """ |
| 28 | |
| 29 | |
| 30 | class ModuleTestCase(unittest.TestCase): |
| 31 | |
| 32 | def test_module(self): |
| 33 | module = interpreter |
| 34 | self.assert_(module.__author__) |
| 35 | self.assert_(module.__cvsid__) |
| 36 | self.assert_(module.__revision__) |
| 37 | self.assert_(module.Interpreter) |
| 38 | self.assert_(module.Interpreter.push) |
| 39 | self.assert_(module.Interpreter.runsource) |
| 40 | self.assert_(module.Interpreter.getAutoCompleteList) |
| 41 | self.assert_(module.Interpreter.getCallTip) |
| 42 | self.assert_(module.InterpreterAlaCarte) |
| 43 | |
| 44 | |
| 45 | class InterpreterTestCase(unittest.TestCase): |
| 46 | |
| 47 | def setUp(self): |
| 48 | self.output = '' |
| 49 | self.i = interpreter.Interpreter(stdout=self) |
| 50 | |
| 51 | def write(self, text): |
| 52 | """Capture output from self.i.push().""" |
| 53 | self.output += text |
| 54 | |
| 55 | def tearDown(self): |
| 56 | self.output = '' |
| 57 | self.i = None |
| 58 | del self.i |
| 59 | |
| 60 | def test_more(self): |
| 61 | self.assertEqual(self.i.push('dir()'), 0) |
| 62 | self.assertEqual(self.i.push('for n in range(3):'), 1) |
| 63 | |
| 64 | def test_push(self): |
| 65 | values = ( |
| 66 | ('dir', '<built-in function dir>'), |
| 67 | ('dir()', "['__builtins__', '__doc__', '__name__']"), |
| 68 | ('2 + 2', '4'), |
| 69 | ('d = {}', ''), |
| 70 | ('d', '{}'), |
| 71 | ('del d', ''), |
| 72 | ('len([4,5,6])', '3'), |
| 73 | ) |
| 74 | for input, output in values: |
| 75 | if output: output += '\n' |
| 76 | self.i.push(input) |
| 77 | self.assertEqual(self.output, output) |
| 78 | self.output = '' |
| 79 | |
| 80 | |
| 81 | if __name__ == '__main__': |
| 82 | unittest.main() |