mirror of
https://github.com/ethereum/solidity
synced 2023-10-03 13:03:40 +00:00
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
import unittest
|
|
|
|
from unittest_helpers import FIXTURE_DIR, load_fixture
|
|
|
|
# NOTE: This test file file only works with scripts/ added to PYTHONPATH so pylint can't find the imports
|
|
# pragma pylint: disable=import-error
|
|
from isolate_tests import extract_docs_cases
|
|
# pragma pylint: enable=import-error
|
|
|
|
|
|
CODE_BLOCK_RST_PATH = FIXTURE_DIR / 'code_block.rst'
|
|
CODE_BLOCK_RST_CONTENT = load_fixture(CODE_BLOCK_RST_PATH)
|
|
CODE_BLOCK_WITH_DIRECTIVES_RST_PATH = FIXTURE_DIR / 'code_block_with_directives.rst'
|
|
CODE_BLOCK_WITH_DIRECTIVES_RST_CONTENT = load_fixture(CODE_BLOCK_WITH_DIRECTIVES_RST_PATH)
|
|
|
|
|
|
class TestExtractDocsCases(unittest.TestCase):
|
|
def setUp(self):
|
|
self.maxDiff = 10000
|
|
|
|
def test_solidity_block(self):
|
|
expected_cases = [
|
|
" // SPDX-License-Identifier: GPL-3.0\n"
|
|
" pragma solidity >=0.7.0 <0.9.0;\n"
|
|
"\n"
|
|
" contract C {\n"
|
|
" function foo() public view {}\n"
|
|
" }\n"
|
|
"\n"
|
|
"\n",
|
|
|
|
" contract C {}\n"
|
|
"\n",
|
|
]
|
|
|
|
self.assertEqual(extract_docs_cases(CODE_BLOCK_RST_PATH), expected_cases)
|
|
|
|
def test_solidity_block_with_directives(self):
|
|
expected_cases = [
|
|
" // SPDX-License-Identifier: GPL-3.0\n"
|
|
" pragma solidity >=0.7.0 <0.9.0;\n"
|
|
"\n"
|
|
" contract C {\n"
|
|
" function foo() public view {}\n"
|
|
" }\n"
|
|
"\n"
|
|
"\n",
|
|
|
|
" contract C {}\n"
|
|
"\n",
|
|
|
|
" contract D {}\n"
|
|
" :linenos:\n"
|
|
"\n",
|
|
|
|
" contract E {}\n"
|
|
"\n",
|
|
]
|
|
|
|
self.assertEqual(extract_docs_cases(CODE_BLOCK_WITH_DIRECTIVES_RST_PATH), expected_cases)
|