|
1 |
| -import unittest |
2 |
| - |
| 1 | +import pytest |
3 | 2 | import mypy.nodes as mpn
|
4 | 3 |
|
5 | 4 | from serializer import typeshed_serializer, symbols
|
6 | 5 |
|
7 | 6 |
|
8 |
| -class TypeshedSerializerTest(unittest.TestCase): |
9 |
| - @classmethod |
10 |
| - def setUpClass(cls) -> None: |
11 |
| - cls.build_result = typeshed_serializer.walk_typeshed_stdlib() |
12 |
| - |
13 |
| - def test_build_mypy_model(self): |
14 |
| - self.assertIsNotNone(self.build_result) |
15 |
| - |
16 |
| - def test_module_symbol(self): |
17 |
| - abc_module = self.build_result.files.get("abc") |
18 |
| - module_symbol = symbols.ModuleSymbol(abc_module) |
19 |
| - self.assertEqual(module_symbol.fullname, "abc") |
20 |
| - self.assertEqual(len(module_symbol.classes), 3) |
21 |
| - self.assertEqual(len(module_symbol.functions), 4) |
22 |
| - |
23 |
| - def test_class_symbol(self): |
24 |
| - mypy_cmd_module = self.build_result.files.get("cmd") |
25 |
| - mypy_cmd_class = mypy_cmd_module.names.get("Cmd") |
26 |
| - cmd_class_symbol = symbols.ClassSymbol(mypy_cmd_class.node) |
27 |
| - self.assertEqual(cmd_class_symbol.fullname, "cmd.Cmd") |
28 |
| - self.assertEqual(cmd_class_symbol.name, "Cmd") |
29 |
| - self.assertEqual(cmd_class_symbol.super_classes, ["builtins.object"]) |
30 |
| - self.assertEqual(len(cmd_class_symbol.mro), 0) |
31 |
| - |
32 |
| - def test_function_symbol(self): |
33 |
| - mypy_cmd_module = self.build_result.files.get("cmd") |
34 |
| - mypy_cmd_class_node = mypy_cmd_module.names.get("Cmd").node |
35 |
| - self.assertIsInstance(mypy_cmd_class_node, mpn.TypeInfo) |
36 |
| - mypy_completenames_method_node = mypy_cmd_class_node.names.get("completenames").node |
37 |
| - self.assertIsInstance(mypy_completenames_method_node, mpn.FuncDef) |
38 |
| - completenames_method_symbol = symbols.FunctionSymbol(mypy_completenames_method_node) |
39 |
| - self.assertEqual(completenames_method_symbol.name, "completenames") |
40 |
| - self.assertEqual(completenames_method_symbol.fullname, "cmd.Cmd.completenames") |
41 |
| - self.assertFalse(completenames_method_symbol.has_decorators) |
42 |
| - self.assertFalse(completenames_method_symbol.is_asynchronous) |
43 |
| - self.assertEqual(completenames_method_symbol.return_type.pretty_printed_name, "builtins.list[builtins.str]") |
44 |
| - self.assertEqual(len(completenames_method_symbol.resolved_decorator_names), 0) |
45 |
| - self.assertEqual(completenames_method_symbol.fullname, "cmd.Cmd.completenames") |
46 |
| - |
47 |
| - args = completenames_method_symbol.parameters |
48 |
| - self.assertEqual(len(args), 3) |
49 |
| - |
50 |
| - |
51 |
| -if __name__ == '__main__': |
52 |
| - unittest.main() |
| 7 | +@pytest.fixture(scope="module") |
| 8 | +def typeshed_stdlib(): |
| 9 | + return typeshed_serializer.walk_typeshed_stdlib() |
| 10 | + |
| 11 | + |
| 12 | +def test_build_mypy_model(typeshed_stdlib): |
| 13 | + assert typeshed_stdlib is not None |
| 14 | + |
| 15 | + |
| 16 | +def test_module_symbol(typeshed_stdlib): |
| 17 | + abc_module = typeshed_stdlib.files.get("abc") |
| 18 | + module_symbol = symbols.ModuleSymbol(abc_module) |
| 19 | + assert module_symbol.fullname == "abc" |
| 20 | + assert len(module_symbol.classes) == 3 |
| 21 | + assert len(module_symbol.functions) == 4 |
| 22 | + |
| 23 | + |
| 24 | +def test_class_symbol(typeshed_stdlib): |
| 25 | + mypy_cmd_module = typeshed_stdlib.files.get("cmd") |
| 26 | + mypy_cmd_class = mypy_cmd_module.names.get("Cmd") |
| 27 | + cmd_class_symbol = symbols.ClassSymbol(mypy_cmd_class.node) |
| 28 | + assert cmd_class_symbol.fullname == "cmd.Cmd" |
| 29 | + assert cmd_class_symbol.name == "Cmd" |
| 30 | + assert cmd_class_symbol.super_classes == ["builtins.object"] |
| 31 | + assert len(cmd_class_symbol.mro) == 0 |
| 32 | + |
| 33 | + |
| 34 | +def test_function_symbol(typeshed_stdlib): |
| 35 | + mypy_cmd_module = typeshed_stdlib.files.get("cmd") |
| 36 | + |
| 37 | + mypy_cmd_class_node = mypy_cmd_module.names.get("Cmd").node |
| 38 | + assert isinstance(mypy_cmd_class_node, mpn.TypeInfo) |
| 39 | + |
| 40 | + mypy_completenames_method_node = mypy_cmd_class_node.names.get("completenames").node |
| 41 | + assert isinstance(mypy_completenames_method_node, mpn.FuncDef) |
| 42 | + |
| 43 | + completenames_method_symbol = symbols.FunctionSymbol(mypy_completenames_method_node) |
| 44 | + assert completenames_method_symbol.name == "completenames" |
| 45 | + assert completenames_method_symbol.fullname == "cmd.Cmd.completenames" |
| 46 | + assert not completenames_method_symbol.has_decorators |
| 47 | + assert not completenames_method_symbol.is_asynchronous |
| 48 | + assert completenames_method_symbol.return_type.pretty_printed_name == "builtins.list[builtins.str]" |
| 49 | + assert len(completenames_method_symbol.resolved_decorator_names) == 0 |
| 50 | + assert completenames_method_symbol.fullname == "cmd.Cmd.completenames" |
| 51 | + |
| 52 | + args = completenames_method_symbol.parameters |
| 53 | + assert len(args) == 3 |
0 commit comments