def test_get_logger_custom_name(self):
logger1 = get_logger("test.module")
self.assertEqual("test.module", logger1.name)
import sys
import time
from typing import Dict, List, Optional
from torchelastic.tsm.driver.api import (
Application,
AppState,
DescribeAppResponse,
RunMode,
Scheduler,
is_terminal,
macros,
)
from torchelastic.utils.logging import get_logger
log = get_logger()
class ImageFetcher(abc.ABC):
"""
Downloads and sets up an image onto the localhost. This is only needed for
``LocalhostScheduler`` since typically real schedulers will do this
on-behalf of the user.
"""
@abc.abstractmethod
def fetch(self, image: str) -> str:
"""
Pulls the given image and returns a path to the pulled image on
the local host.
"""
raise NotImplementedError()
def test_get_logger_none(self):
logger1 = get_logger(None)
self.assertEqual(__name__, logger1.name)
def test_get_logger(self):
logger1 = get_logger()
self.assertEqual(__name__, logger1.name)
def test_get_logger_different(self):
logger1 = get_logger("name1")
logger2 = get_logger("name2")
self.assertNotEqual(logger1.name, logger2.name)
def test_get_logger_same(self):
logger1 = get_logger(__name__)
logger2 = get_logger(__name__)
self.assertEqual(1, len(logger1.handlers))
self.assertEqual(logger1, logger2)
#!/usr/bin/env python3
# Copyright (c) Facebook, Inc. and its affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
import unittest
import torchelastic.utils.logging as logging
log = logging.get_logger()
class LoggingTest(unittest.TestCase):
def test_logger_name(self):
self.assertEqual("torchelastic.utils.test.logging_test", log.name)
def test_derive_module_name(self):
module_name = logging._derive_module_name(depth=1)
self.assertEqual("torchelastic.utils.test.logging_test", module_name)
