upstream/ipython Files · IPython/extensions/tests/test_cythonmagic.py

eliminate hardcoded wait, now wait on outputs...

eliminate hardcoded wait, now wait on outputs Note: this test still sometimes fails, though I don't understand why. Increasing the wait time to 30 seconds does not help, which leads me to believe that there's some race condition, or that we're genuinely dropping outputs sometimes (saved notebooks on these timeouts *do* contain an In[] number, but don't have any outputs attached). @ellisonbg and @minrk might now what's going on with that. To run just this test, fire up a notebook server on port 8888 and run: while true; do casperjs test --includes=util.js test_cases/execute_code_cell.js ; done

Cyrille Rossant - - Load All Authors

File last commit:

r12647:0e0e6a8e


                r13277:8cc2a54f

Download file

             test_cythonmagic.py
        
                    70 lines
            
             | 1.6 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / extensions / tests / test_cythonmagic.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # -*- coding: utf-8 -*-

      """Tests for the Cython magics extension."""

      import os

      import nose.tools as nt

      from IPython.testing import decorators as dec

      from IPython.utils import py3compat

      code = py3compat.str_to_unicode("""def f(x):

          return 2*x

      """)

      try:

          import Cython

      except:

          __test__ = False

      ip = get_ipython()

      def setup():

          ip.extension_manager.load_extension('cythonmagic')

      def test_cython_inline():

          ip.ex('a=10; b=20')

          result = ip.run_cell_magic('cython_inline','','return a+b')

          nt.assert_equal(result, 30)

      @dec.skip_win32

      def test_cython_pyximport():

          module_name = '_test_cython_pyximport'

          ip.run_cell_magic('cython_pyximport', module_name, code)

          ip.ex('g = f(10)')

          nt.assert_equal(ip.user_ns['g'], 20.0)

          ip.run_cell_magic('cython_pyximport', module_name, code)

          ip.ex('h = f(-10)')

          nt.assert_equal(ip.user_ns['h'], -20.0)

          try:

              os.remove(module_name+'.pyx')

          except OSError:

              pass

      def test_cython():

          ip.run_cell_magic('cython', '', code)

          ip.ex('g = f(10)')

          nt.assert_equal(ip.user_ns['g'], 20.0)

      def test_cython_name():

          # The Cython module named 'mymodule' defines the function f.

          ip.run_cell_magic('cython', '--name=mymodule', code)

          # This module can now be imported in the interactive namespace.

          ip.ex('import mymodule; g = mymodule.f(10)')

          nt.assert_equal(ip.user_ns['g'], 20.0)

      @dec.skip_win32

      def test_extlibs():

          code = py3compat.str_to_unicode("""

      from libc.math cimport sin

      x = sin(0.0)

          """)

          ip.user_ns['x'] = 1

          ip.run_cell_magic('cython', '-l m', code)

          nt.assert_equal(ip.user_ns['x'], 0)

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				# -- coding: utf-8 --
				"""Tests for the Cython magics extension."""

				import os
				import nose.tools as nt

				from IPython.testing import decorators as dec
				from IPython.utils import py3compat

				code = py3compat.str_to_unicode("""def f(x):
				return 2*x
				""")

				try:
				import Cython
				except:
				__test__ = False

				ip = get_ipython()


				def setup():
				ip.extension_manager.load_extension('cythonmagic')


				def test_cython_inline():
				ip.ex('a=10; b=20')
				result = ip.run_cell_magic('cython_inline','','return a+b')
				nt.assert_equal(result, 30)


				@dec.skip_win32
				def test_cython_pyximport():
				module_name = '_test_cython_pyximport'
				ip.run_cell_magic('cython_pyximport', module_name, code)
				ip.ex('g = f(10)')
				nt.assert_equal(ip.user_ns['g'], 20.0)
				ip.run_cell_magic('cython_pyximport', module_name, code)
				ip.ex('h = f(-10)')
				nt.assert_equal(ip.user_ns['h'], -20.0)
				try:
				os.remove(module_name+'.pyx')
				except OSError:
				pass


				def test_cython():
				ip.run_cell_magic('cython', '', code)
				ip.ex('g = f(10)')
				nt.assert_equal(ip.user_ns['g'], 20.0)


				def test_cython_name():
				# The Cython module named 'mymodule' defines the function f.
				ip.run_cell_magic('cython', '--name=mymodule', code)
				# This module can now be imported in the interactive namespace.
				ip.ex('import mymodule; g = mymodule.f(10)')
				nt.assert_equal(ip.user_ns['g'], 20.0)


				@dec.skip_win32
				def test_extlibs():
				code = py3compat.str_to_unicode("""
				from libc.math cimport sin
				x = sin(0.0)
				""")
				ip.user_ns['x'] = 1
				ip.run_cell_magic('cython', '-l m', code)
				nt.assert_equal(ip.user_ns['x'], 0)