mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
29b9c44c0a
Summary: The conda build process generates some files of its own, which we don't want to catch in our test for copyright notices. Reviewed By: nikhilaravi, patricklabatut Differential Revision: D20868566 fbshipit-source-id: 76a786a3eb9a674d59e630cc06f346e8b82258a4
76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
|
|
import unittest
|
|
from collections import Counter
|
|
from pathlib import Path
|
|
|
|
|
|
# This file groups together tests which look at the code without running it.
|
|
|
|
|
|
class TestBuild(unittest.TestCase):
|
|
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.
|
|
test_dir = Path(__file__).resolve().parent
|
|
source_dir = test_dir.parent / "pytorch3d"
|
|
|
|
stems = []
|
|
for extension in [".cu", ".cpp"]:
|
|
files = source_dir.glob(f"**/*{extension}")
|
|
stems.extend(f.stem for f in files)
|
|
|
|
counter = Counter(stems)
|
|
for k, v in counter.items():
|
|
self.assertEqual(v, 1, f"Too many files with stem {k}.")
|
|
|
|
def test_deprecated_usage(self):
|
|
# Check certain expressions do not occur in the csrc code
|
|
test_dir = Path(__file__).resolve().parent
|
|
source_dir = test_dir.parent / "pytorch3d" / "csrc"
|
|
|
|
files = sorted(source_dir.glob("**/*.*"))
|
|
self.assertGreater(len(files), 4)
|
|
|
|
patterns = [".type()", ".data()"]
|
|
|
|
for file in files:
|
|
with open(file) as f:
|
|
text = f.read()
|
|
for pattern in patterns:
|
|
found = pattern in text
|
|
msg = (
|
|
f"{pattern} found in {file.name}"
|
|
+ ", this has been deprecated."
|
|
)
|
|
self.assertFalse(found, msg)
|
|
|
|
def test_copyright(self):
|
|
test_dir = Path(__file__).resolve().parent
|
|
root_dir = test_dir.parent
|
|
|
|
extensions = ("py", "cu", "cuh", "cpp", "h", "hpp", "sh")
|
|
|
|
expect = (
|
|
"Copyright (c) Facebook, Inc. and its affiliates."
|
|
+ " All rights reserved.\n"
|
|
)
|
|
|
|
conda_generated_files = [
|
|
"run_test.py",
|
|
"run_test.sh",
|
|
"conda_test_runner.sh",
|
|
"conda_test_env_vars.sh",
|
|
]
|
|
|
|
for extension in extensions:
|
|
for i in root_dir.glob(f"**/*.{extension}"):
|
|
if i.name in conda_generated_files:
|
|
continue
|
|
with open(i) as f:
|
|
firstline = f.readline()
|
|
if firstline.startswith(("# -*-", "#!")):
|
|
firstline = f.readline()
|
|
self.assertTrue(
|
|
firstline.endswith(expect), f"{i} missing copyright header."
|
|
)
|