| import torch | |
| from common_utils import TestCase, run_tests | |
| class LoggingTest(TestCase): | |
| def testApiUsage(self): | |
| """ | |
| This test verifies that api usage logging is not triggered via static | |
| initialization. Since it's triggered at first invocation only - we just | |
| subprocess | |
| """ | |
| s = TestCase.runWithPytorchAPIUsageStderr("import torch") | |
| self.assertRegexpMatches(s, "PYTORCH_API_USAGE.*import") | |
| # import the shared library directly - it triggers static init but doesn't call anything | |
| s = TestCase.runWithPytorchAPIUsageStderr("from ctypes import CDLL; CDLL('{}')".format(torch._C.__file__)) | |
| self.assertNotRegexpMatches(s, "PYTORCH_API_USAGE") | |
| if __name__ == '__main__': | |
| run_tests() |