test_build

Summary: Ensure copyright header consistency and translation unit name uniqueness.

Reviewed By: nikhilaravi

Differential Revision: D20438802

fbshipit-source-id: 9820cfe4c6efab016a0a8589dfa24bb526692f83

tests/bm_blending.py

@@ -1,4 +1,4 @@
-# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 
 from itertools import product

tests/bm_mesh_laplacian_smoothing.py

@@ -1,4 +1,4 @@
-# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 
 from itertools import product

tests/bm_mesh_normal_consistency.py

@@ -1,4 +1,4 @@
-# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 
 from itertools import product

tests/test_build.py

@@ -0,0 +1,45 @@
+# 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_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"
+        )
+
+        for extension in extensions:
+            for i in root_dir.glob(f"**/*.{extension}"):
+                with open(i) as f:
+                    firstline = f.readline()
+                    if firstline.startswith(("# -*-", "#!")):
+                        firstline = f.readline()
+                    self.assertTrue(
+                        firstline.endswith(expect),
+                        f"{i} missing copyright header.",
+                    )

tests/test_mesh_laplacian_smoothing.py

@@ -1,4 +1,4 @@
-# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 
 import unittest

tests/test_mesh_normal_consistency.py

@@ -1,4 +1,4 @@
-# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 
 import unittest

tests/test_rotation_conversions.py

@@ -1,4 +1,4 @@
-# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 
 import itertools