mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-07-31 10:52:50 +08:00
e55ea90609
Summary: these tests don't work, aren't needed right now Reviewed By: MichaelRamamonjisoa Differential Revision: D78084742 fbshipit-source-id: 9cff2b30427dec314e34e81179816af4073bbe23
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
import importlib
|
|
import os
|
|
import sys
|
|
import unittest
|
|
import unittest.mock
|
|
|
|
from tests.common_testing import get_pytorch3d_dir
|
|
|
|
|
|
# This file groups together tests which look at the code without running it.
|
|
class TestBuild(unittest.TestCase):
|
|
def _test_no_import_cycles(self):
|
|
# Check each module of pytorch3d imports cleanly,
|
|
# which may fail if there are import cycles.
|
|
|
|
with unittest.mock.patch.dict(sys.modules):
|
|
for module in list(sys.modules):
|
|
# If any of pytorch3d is already imported,
|
|
# the test would be pointless.
|
|
if module.startswith("pytorch3d"):
|
|
sys.modules.pop(module, None)
|
|
|
|
# torchvision seems to cause problems if re-imported,
|
|
# so make sure it has been imported here.
|
|
import torchvision.utils # noqa
|
|
|
|
root_dir = get_pytorch3d_dir() / "pytorch3d"
|
|
# Exclude opengl-related files, as Implicitron is decoupled from opengl
|
|
# components which will not work without adding a dep on pytorch3d_opengl.
|
|
ignored_modules = (
|
|
"__init__",
|
|
"plotly_vis",
|
|
"opengl_utils",
|
|
"rasterizer_opengl",
|
|
)
|
|
if os.environ.get("FB_TEST", False):
|
|
ignored_modules += ("orm_types", "sql_dataset", "sql_dataset_provider")
|
|
for module_file in root_dir.glob("**/*.py"):
|
|
if module_file.stem in ignored_modules:
|
|
continue
|
|
relative_module = str(module_file.relative_to(root_dir))[:-3]
|
|
module = "pytorch3d." + relative_module.replace("/", ".")
|
|
with self.subTest(name=module):
|
|
with unittest.mock.patch.dict(sys.modules):
|
|
importlib.import_module(module)
|