1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
# 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.
from pathlib import Path
from unittest import TestCase
from libcst import CSTTransformer, parse_module
fixtures: Path = Path(__file__).parent.parent.parent / "native/libcst/tests/fixtures"
class NOOPTransformer(CSTTransformer):
pass
class RoundTripTests(TestCase):
def _get_fixtures(self) -> list[Path]:
self.assertTrue(fixtures.exists(), f"{fixtures} should exist")
files = list(fixtures.iterdir())
self.assertGreater(len(files), 0)
return files
def test_clean_roundtrip(self) -> None:
for file in self._get_fixtures():
with self.subTest(file=str(file)):
src = file.read_text(encoding="utf-8")
mod = parse_module(src)
self.maxDiff = None
self.assertEqual(mod.code, src)
def test_transform_roundtrip(self) -> None:
transformer = NOOPTransformer()
self.maxDiff = None
for file in self._get_fixtures():
with self.subTest(file=str(file)):
src = file.read_text(encoding="utf-8")
mod = parse_module(src)
new_mod = mod.visit(transformer)
self.assertEqual(src, new_mod.code)
|
