mirror of
https://github.com/Instagram/LibCST.git
synced 2025-12-23 10:35:53 +00:00
7ca1bd1cd5
* expression: fix SimpleString's quote method * Add missing copyright header Co-authored-by: zzl0 <zhuzhaolong0@mail.com> Co-authored-by: Luke Petre <lpetre@fb.com>
31 lines
880 B
Python
31 lines
880 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
import unittest
|
|
|
|
import libcst as cst
|
|
|
|
|
|
class TestSimpleString(unittest.TestCase):
|
|
def test_quote(self) -> None:
|
|
test_cases = [
|
|
('"a"', '"'),
|
|
("'b'", "'"),
|
|
('""', '"'),
|
|
("''", "'"),
|
|
('"""c"""', '"""'),
|
|
("'''d'''", "'''"),
|
|
('""""e"""', '"""'),
|
|
("''''f'''", "'''"),
|
|
('"""""g"""', '"""'),
|
|
("'''''h'''", "'''"),
|
|
('""""""', '"""'),
|
|
("''''''", "'''"),
|
|
]
|
|
|
|
for s, expected_quote in test_cases:
|
|
simple_string = cst.SimpleString(s)
|
|
actual = simple_string.quote
|
|
self.assertEqual(expected_quote, actual)
|