mirror of
https://github.com/Instagram/LibCST.git
synced 2025-12-23 10:35:53 +00:00
Standardize on the convention that private modules (those we don't expect people to directly import) are prefixed with an underscore. Everything under a directory/module that has an underscore is considered private, unless it is re-exported from a non-underscored module. Most things are exported from libcst directly, but there are a few things in libcst.tool, libcst.codegen and libcst.metadata that are namedspaced as such.
45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
from libcst._parser.grammar import _should_include
|
|
from libcst._parser.parso.utils import PythonVersionInfo
|
|
from libcst.testing.utils import UnitTest, data_provider
|
|
|
|
|
|
class VersionCompareTest(UnitTest):
|
|
@data_provider(
|
|
(
|
|
# Simple equality
|
|
("==3.6", PythonVersionInfo(3, 6), True),
|
|
("!=3.6", PythonVersionInfo(3, 6), False),
|
|
# Equal or GT/LT
|
|
(">=3.6", PythonVersionInfo(3, 5), False),
|
|
(">=3.6", PythonVersionInfo(3, 6), True),
|
|
(">=3.6", PythonVersionInfo(3, 7), True),
|
|
("<=3.6", PythonVersionInfo(3, 5), True),
|
|
("<=3.6", PythonVersionInfo(3, 6), True),
|
|
("<=3.6", PythonVersionInfo(3, 7), False),
|
|
# GT/LT
|
|
(">3.6", PythonVersionInfo(3, 5), False),
|
|
(">3.6", PythonVersionInfo(3, 6), False),
|
|
(">3.6", PythonVersionInfo(3, 7), True),
|
|
("<3.6", PythonVersionInfo(3, 5), True),
|
|
("<3.6", PythonVersionInfo(3, 6), False),
|
|
("<3.6", PythonVersionInfo(3, 7), False),
|
|
# Multiple checks
|
|
(">3.6,<3.8", PythonVersionInfo(3, 6), False),
|
|
(">3.6,<3.8", PythonVersionInfo(3, 7), True),
|
|
(">3.6,<3.8", PythonVersionInfo(3, 8), False),
|
|
)
|
|
)
|
|
def test_tokenize(
|
|
self,
|
|
requested_version: str,
|
|
actual_version: PythonVersionInfo,
|
|
expected_result: bool,
|
|
) -> None:
|
|
self.assertEqual(
|
|
_should_include(requested_version, actual_version), expected_result
|
|
)
|