mirror of
https://github.com/readthedocs/sphinx-autoapi
synced 2024-11-11 19:10:58 +00:00
69 lines
1.6 KiB
Python
69 lines
1.6 KiB
Python
|
# coding=utf8
|
||
|
|
|
||
|
|
"""Test Python parser"""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import unittest
|
||
|
|
from textwrap import dedent
|
||
|
|
|
||
|
|
from autoapi.mappers.python import ParserExtra
|
||
|
|
|
||
|
|
if sys.version_info < (3, 0):
|
||
|
|
from StringIO import StringIO
|
||
|
|
else:
|
||
|
|
from io import StringIO
|
||
|
|
|
||
|
|
|
||
|
|
class PythonParserTests(unittest.TestCase):
|
||
|
|
|
||
|
|
def parse(self, source):
|
||
|
|
in_h = StringIO(dedent(source))
|
||
|
|
return ParserExtra()(in_h, '/dev/null')
|
||
|
|
|
||
|
|
def test_parses_basic_file(self):
|
||
|
|
source = """
|
||
|
|
def foo(bar):
|
||
|
|
pass
|
||
|
|
"""
|
||
|
|
self.assertIsNone(self.parse(source).all)
|
||
|
|
|
||
|
|
def test_parses_all(self):
|
||
|
|
source = """
|
||
|
|
__all__ = ['Foo', 5.0]
|
||
|
|
"""
|
||
|
|
self.assertEqual(self.parse(source).all, ['Foo', 5.0])
|
||
|
|
|
||
|
|
def test_parses_all_with_list_addition(self):
|
||
|
|
source = """
|
||
|
|
__all__ = ['Foo'] + []
|
||
|
|
"""
|
||
|
|
self.assertEqual(self.parse(source).all, ['Foo'])
|
||
|
|
|
||
|
|
def test_parses_all_with_name_addtion(self):
|
||
|
|
source = """
|
||
|
|
__all__ = ['Foo'] + bar.__all__
|
||
|
|
"""
|
||
|
|
self.assertEqual(self.parse(source).all, ['Foo'])
|
||
|
|
|
||
|
|
def test_parses_all_with_multiple_name_addtions(self):
|
||
|
|
source = """
|
||
|
|
__all__ = foo + bar
|
||
|
|
__all__ += boop
|
||
|
|
__all__ += ['foo']
|
||
|
|
"""
|
||
|
|
self.assertEqual(self.parse(source).all, ['foo'])
|
||
|
|
source = """
|
||
|
|
__all__ = ['foo']
|
||
|
|
__all__ = foo
|
||
|
|
"""
|
||
|
|
self.assertEqual(self.parse(source).all, [])
|
||
|
|
|
||
|
|
def test_parses_all_multiline(self):
|
||
|
|
source = """
|
||
|
|
__all__ = [
|
||
|
|
'foo',
|
||
|
|
'bar',
|
||
|
|
]
|
||
|
|
"""
|
||
|
|
self.assertEqual(self.parse(source).all, ['foo', 'bar'])
|