upstream/ipython Files · IPython/utils/tests/test_dir2.py

Improve async detection mechanism with blacklist...

Improve async detection mechanism with blacklist Because the async repl works by wrapping any code that raises SyntaxError in an async function and trying to execute it again, cell bodies that are invalid at the top level but valid in functions and methods (e.g. return and yield statements) currently allow executing invalid code. This patch blacklists return and yield statements outside of a function or method to restore the proper SyntaxError behavior.

Thomas Kluyver - - Load All Authors

File last commit:

r21287:23cd8cd1


                r24485:c4bf6326

Download file

             test_dir2.py
        
                    58 lines
            
             | 1.3 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / utils / tests / test_dir2.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      import nose.tools as nt

      from IPython.utils.dir2 import dir2

      class Base(object):

          x = 1

          z = 23

      def test_base():

          res = dir2(Base())

          assert ('x' in res)

          assert ('z' in res)

          assert ('y' not in res)

          assert ('__class__' in res)

          nt.assert_equal(res.count('x'), 1)

          nt.assert_equal(res.count('__class__'), 1)

      def test_SubClass():

          class SubClass(Base):

              y = 2

          res = dir2(SubClass())

          assert ('y' in res)

          nt.assert_equal(res.count('y'), 1)

          nt.assert_equal(res.count('x'), 1)

      def test_SubClass_with_trait_names_attr():

          # usecase: trait_names is used in a class describing psychological classification

          class SubClass(Base):

              y = 2

              trait_names = 44

          res = dir2(SubClass())

          assert('trait_names' in res)

      def test_misbehaving_object_without_trait_names():

          # dir2 shouldn't raise even when objects are dumb and raise

          # something other than AttribteErrors on bad getattr.

          class MisbehavingGetattr(object):

              def __getattr__(self):

                  raise KeyError("I should be caught")

              def some_method(self):

                  pass

          class SillierWithDir(MisbehavingGetattr):

              def __dir__(self):

                  return ['some_method']

          for bad_klass in (MisbehavingGetattr, SillierWithDir):

              res = dir2(bad_klass())

              assert('some_method' in res)

	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

				import nose.tools as nt
				from IPython.utils.dir2 import dir2


				class Base(object):
				x = 1
				z = 23


				def test_base():
				res = dir2(Base())
				assert ('x' in res)
				assert ('z' in res)
				assert ('y' not in res)
				assert ('__class__' in res)
				nt.assert_equal(res.count('x'), 1)
				nt.assert_equal(res.count('__class__'), 1)

				def test_SubClass():

				class SubClass(Base):
				y = 2

				res = dir2(SubClass())
				assert ('y' in res)
				nt.assert_equal(res.count('y'), 1)
				nt.assert_equal(res.count('x'), 1)


				def test_SubClass_with_trait_names_attr():
				# usecase: trait_names is used in a class describing psychological classification

				class SubClass(Base):
				y = 2
				trait_names = 44

				res = dir2(SubClass())
				assert('trait_names' in res)


				def test_misbehaving_object_without_trait_names():
				# dir2 shouldn't raise even when objects are dumb and raise
				# something other than AttribteErrors on bad getattr.

				class MisbehavingGetattr(object):
				def __getattr__(self):
				raise KeyError("I should be caught")

				def some_method(self):
				pass

				class SillierWithDir(MisbehavingGetattr):
				def __dir__(self):
				return ['some_method']

				for bad_klass in (MisbehavingGetattr, SillierWithDir):
				res = dir2(bad_klass())
				assert('some_method' in res)