#!/usr/bin/env python
"""Test suite for the irunner module.

Not the most elegant or fine-grained, but it does cover at least the bulk
functionality."""

# Global to make tests extra verbose and help debugging
VERBOSE = True

# stdlib imports
import cStringIO as StringIO
import unittest

# IPython imports
from IPython import irunner
from IPython.OutputTrap import OutputTrap

# Testing code begins
class RunnerTestCase(unittest.TestCase):

    def _test_runner(self,runner,source,output):
        """Test that a given runner's input/output match."""
        
        log = OutputTrap(out_head='',quiet_out=True)
        log.trap_out()
        runner.run_source(source)
        log.release_out()
        out = log.summary_out()
        # this output contains nasty \r\n lineends, and the initial ipython
        # banner.  clean it up for comparison
        output_l = output.split()
        out_l = out.split()
        mismatch  = 0
        for n in range(len(output_l)):
            ol1 = output_l[n].strip()
            ol2 = out_l[n].strip()
            if ol1 != ol2:
                mismatch += 1
                if VERBOSE:
                    print '<<< line %s does not match:' % n
                    print repr(ol1)
                    print repr(ol2)
                    print '>>>'
        self.assert_(mismatch==0,'Number of mismatched lines: %s' %
                     mismatch)

    def testIPython(self):
        """Test the IPython runner."""
        source = """
print 'hello, this is python'

# some more code
x=1;y=2
x+y**2

# An example of autocall functionality
from math import *
autocall 1
cos pi
autocall 0
cos pi
cos(pi)

for i in range(5):
    print i,

print "that's all folks!"

%Exit
"""
        output = """\
In [1]: print 'hello, this is python'
hello, this is python

In [2]: # some more code

In [3]: x=1;y=2

In [4]: x+y**2
Out[4]: 5

In [5]: # An example of autocall functionality

In [6]: from math import *

In [7]: autocall 1
Automatic calling is: Smart

In [8]: cos pi
------> cos(pi)
Out[8]: -1.0

In [9]: autocall 0
Automatic calling is: OFF

In [10]: cos pi
------------------------------------------------------------
   File "<ipython console>", line 1
     cos pi
          ^
SyntaxError: invalid syntax


In [11]: cos(pi)
Out[11]: -1.0

In [12]: for i in range(5):
   ....:         print i,
   ....: 
0 1 2 3 4

In [13]: print "that's all folks!"
that's all folks!

In [14]: %Exit"""
        runner = irunner.IPythonRunner()
        self._test_runner(runner,source,output)

    def testPython(self):
        """Test the Python runner."""
        runner = irunner.PythonRunner()
        source = """
print 'hello, this is python'

# some more code
x=1;y=2
x+y**2

from math import *
cos(pi)

for i in range(5):
    print i,

print "that's all folks!"
        """
        output = """\
>>> print 'hello, this is python'
hello, this is python
>>> # some more code
... x=1;y=2
>>> x+y**2
5
>>> from math import *
>>> cos(pi)
-1.0
>>> for i in range(5):
...     print i,
...
0 1 2 3 4
>>> print "that's all folks!"
that's all folks!"""
        self._test_runner(runner,source,output)
        
if __name__ == '__main__':
    unittest.main()