mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
ba48cf6535
See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501. Most changes are auto-generated by linter. You can review these PRs via: ```bash git diff --ignore-all-space --ignore-blank-lines HEAD~1 ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/129757 Approved by: https://github.com/ezyang
44 lines
1.0 KiB
Python
44 lines
1.0 KiB
Python
# Owner(s): ["oncall: package/deploy"]
|
|
|
|
import textwrap
|
|
import types
|
|
|
|
from torch.testing._internal.common_utils import run_tests, TestCase
|
|
from torch.utils._freeze import Freezer, PATH_MARKER
|
|
|
|
|
|
class TestFreezer(TestCase):
|
|
"""Tests the freeze.py script"""
|
|
|
|
def test_compile_string(self):
|
|
freezer = Freezer(True)
|
|
code_str = textwrap.dedent(
|
|
"""
|
|
class MyCls:
|
|
def __init__(self):
|
|
pass
|
|
"""
|
|
)
|
|
co = freezer.compile_string(code_str)
|
|
num_co = 0
|
|
|
|
def verify_filename(co: types.CodeType):
|
|
nonlocal num_co
|
|
|
|
if not isinstance(co, types.CodeType):
|
|
return
|
|
|
|
self.assertEqual(PATH_MARKER, co.co_filename)
|
|
num_co += 1
|
|
|
|
for nested_co in co.co_consts:
|
|
verify_filename(nested_co)
|
|
|
|
verify_filename(co)
|
|
# there is at least one nested code object besides the top level one
|
|
self.assertTrue(num_co >= 2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|