"docs/en/user_guides/config.md" did not exist on "2c29afa3ff69c7dbf5aa6274549f53bd89abf9c0"
test_build.py 2.66 KB
Newer Older
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
1
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
2
import json
3
import os
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
4
5
import unittest
from collections import Counter
6
7

from common_testing import get_pytorch3d_dir, get_tests_dir
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
8

9

Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
10
# This file groups together tests which look at the code without running it.
11
12
# When running the tests inside conda's build, the code is not available.
in_conda_build = os.environ.get("CONDA_BUILD_STATE", "") == "TEST"
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
13
14
15


class TestBuild(unittest.TestCase):
16
    @unittest.skipIf(in_conda_build, "In conda build")
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
17
18
19
    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.
20
        test_dir = get_tests_dir()
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
21
22
23
24
25
26
27
28
29
30
31
        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}.")

32
    @unittest.skipIf(in_conda_build, "In conda build")
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
33
    def test_copyright(self):
34
        test_dir = get_tests_dir()
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
35
36
37
38
39
40
41
42
43
        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"
        )

Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
44
45
        files_missing_copyright_header = []

Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
46
        for extension in extensions:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
47
48
            for path in root_dir.glob(f"**/*.{extension}"):
                if str(path).endswith(
Christoph Lassner's avatar
Christoph Lassner committed
49
50
51
                    "pytorch3d/transforms/external/kornia_angle_axis_to_rotation_matrix.py"
                ):
                    continue
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
52
                if str(path).endswith("pytorch3d/csrc/pulsar/include/fastermath.h"):
Christoph Lassner's avatar
Christoph Lassner committed
53
                    continue
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
54
                with open(path) as f:
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
55
56
57
                    firstline = f.readline()
                    if firstline.startswith(("# -*-", "#!")):
                        firstline = f.readline()
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
58
59
60
61
62
63
64
65
66
                    if not firstline.endswith(expect):
                        files_missing_copyright_header.append(str(path))

        if len(files_missing_copyright_header) != 0:
            self.fail("\n".join(files_missing_copyright_header))

    @unittest.skipIf(in_conda_build, "In conda build")
    def test_valid_ipynbs(self):
        # Check that the ipython notebooks are valid json
67
68
        root_dir = get_pytorch3d_dir()
        tutorials_dir = root_dir / "docs" / "tutorials"
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
69
70
71
72
73
        tutorials = sorted(tutorials_dir.glob("*.ipynb"))

        for tutorial in tutorials:
            with open(tutorial) as f:
                json.load(f)