423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-20 22:30:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allow tests to be run on GPU with remote execution

Browse Source 
Summary: Test path special case

Reviewed By: bottler

Differential Revision: D27566817

fbshipit-source-id: c7b3ac839908c071f1378a37b7013b91ca4e8b18
This commit is contained in:
Rong Rong (AI Infra)
2021-04-08 20:02:16 -07:00
committed by Facebook GitHub Bot
parent dd8343922e
commit c2e62a5087
2 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_build.py
View File

@@ -10,10 +10,11 @@ from common_testing import get_pytorch3d_dir, get_tests_dir
# This file groups together tests which look at the code without running it.
# When running the tests inside conda's build, the code is not available.
in_conda_build = os.environ.get("CONDA_BUILD_STATE", "") == "TEST"
in_re_worker = os.environ.get("INSIDE_RE_WORKER", "") is not None
class TestBuild(unittest.TestCase):
@unittest.skipIf(in_conda_build, "In conda build")
@unittest.skipIf(in_conda_build or in_re_worker, "In conda build, or RE worker")
def test_name_clash(self):
# For setup.py, all translation units need distinct names, so we
# cannot have foo.cu and foo.cpp, even in different directories.
@@ -29,7 +30,7 @@ class TestBuild(unittest.TestCase):
for k, v in counter.items():
self.assertEqual(v, 1, f"Too many files with stem {k}.")
@unittest.skipIf(in_conda_build, "In conda build")
@unittest.skipIf(in_conda_build or in_re_worker, "In conda build, or RE worker")
def test_copyright(self):
test_dir = get_tests_dir()
root_dir = test_dir.parent
@@ -61,7 +62,7 @@ class TestBuild(unittest.TestCase):
if len(files_missing_copyright_header) != 0:
self.fail("\n".join(files_missing_copyright_header))
@unittest.skipIf(in_conda_build, "In conda build")
@unittest.skipIf(in_conda_build or in_re_worker, "In conda build, or RE worker")
def test_valid_ipynbs(self):
# Check that the ipython notebooks are valid json
root_dir = get_pytorch3d_dir()