423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 11:52:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

update test_build for robustness

Browse Source 
Summary: Change cyclic deps test to be independent of test discovery order. Also let it work without plotly.

Reviewed By: nikhilaravi

Differential Revision: D30669614

fbshipit-source-id: 2eadf3f8b56b6096c5466ce53b4f8ac6df27b964
This commit is contained in:
Jeremy Reizenstein 2021-09-02 08:38:10 -07:00 committed by Facebook GitHub Bot
parent a9b0d50baf
commit f2c44e3540
1 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tests/test_build.py
View File

@ -118,17 +118,19 @@ class TestBuild(unittest.TestCase):
# Check each module of pytorch3d imports cleanly, # Check each module of pytorch3d imports cleanly,
# which may fail if there are import cycles. # which may fail if there are import cycles.
# First check the setup of the test. If any of pytorch3d with unittest.mock.patch.dict(sys.modules):
# was already imported the test would be pointless. for module in list(sys.modules):
for module in sys.modules: # If any of pytorch3d is already imported,
self.assertFalse(module.startswith("pytorch3d"), module) # the test would be pointless.
if module.startswith("pytorch3d"):
sys.modules.pop(module, None)
root_dir = get_pytorch3d_dir() / "pytorch3d" root_dir = get_pytorch3d_dir() / "pytorch3d"
for module_file in root_dir.glob("**/*.py"): for module_file in root_dir.glob("**/*.py"):
if module_file.stem == "__init__": if module_file.stem in ("__init__", "plotly_vis"):
continue continue
relative_module = str(module_file.relative_to(root_dir))[:-3] relative_module = str(module_file.relative_to(root_dir))[:-3]
module = "pytorch3d." + relative_module.replace("/", ".") module = "pytorch3d." + relative_module.replace("/", ".")
with self.subTest(name=module): with self.subTest(name=module):
with unittest.mock.patch.dict(sys.modules): with unittest.mock.patch.dict(sys.modules):
importlib.import_module(module) importlib.import_module(module)