test_build

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

Reviewed By: nikhilaravi

Differential Revision: D20438802

fbshipit-source-id: 9820cfe4c6efab016a0a8589dfa24bb526692f83

docs/conf.py

 # -*- coding: utf-8 -*-
-# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 # flake8: noqa
 

pytorch3d/__init__.py

-# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 __version__ = "0.1.1"

pytorch3d/loss/mesh_laplacian_smoothing.py

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

pytorch3d/loss/mesh_normal_consistency.py

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

pytorch3d/utils/torus.py

-# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 from itertools import tee
 from math import cos, pi, sin

scripts/build_website.sh

 #!/bin/bash
-# Copyright (c) Facebook, Inc. and its affiliates.
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 # run this script from the project root using `./scripts/build_docs.sh`
 
@@ -57,4 +57,4 @@ else
   echo "Starting local server"
   echo "-----------------------------------"
   yarn start
-fi
+fi
\ No newline at end of file

scripts/parse_tutorials.py

 #!/usr/bin/env python3
-# Copyright (c) Facebook, Inc. and its affiliates.
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 import argparse
 import json

scripts/publish_website.sh

 #!/bin/bash
-# Copyright (c) Facebook, Inc. and its affiliates.
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
 usage() {
   echo "Usage: $0 [-b]"

tests/bm_blending.py

-# 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

-# 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

-# 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

+# 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

-# 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

-# 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

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