pytorch

1
# Owner(s): ["oncall: distributed"]
2

3
import logging
4

5
from torch.distributed._shard.sharded_tensor.logger import _get_or_create_logger
6
from torch.testing._internal.common_utils import run_tests, TestCase
7

8

9
class ShardingSpecLoggerTest(TestCase):
10
    def test_get_or_create_logger(self):
11
        logger = _get_or_create_logger()
12
        self.assertIsNotNone(logger)
13
        self.assertEqual(1, len(logger.handlers))
14
        self.assertIsInstance(logger.handlers[0], logging.NullHandler)
15

16

17
if __name__ == "__main__":
18
    run_tests()
19