/
vshmidt
/
pytorch
Обзор
Документация
Войти
/
vshmidt
/
pytorch
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
tools/test/test_utils.py
22 строки
870 B
Mengwei Liu
[torchgen] Support multiple namespace in NativeFunctions.h (#79733)
09 июл 2022, 00:56
09 июл 2022, 00:56
5c8a980
Код
Авторство
О чём код?
import unittest from torchgen.utils import NamespaceHelper class TestNamespaceHelper(unittest.TestCase): def test_create_from_namespaced_tuple(self) -> None: helper = NamespaceHelper.from_namespaced_entity("aten::add") self.assertEqual(helper.entity_name, "add") self.assertEqual(helper.get_cpp_namespace(), "aten") def test_default_namespace(self) -> None: helper = NamespaceHelper.from_namespaced_entity("add") self.assertEqual(helper.entity_name, "add") self.assertEqual(helper.get_cpp_namespace(), "") self.assertEqual(helper.get_cpp_namespace("default"), "default") def test_namespace_levels_more_than_max(self) -> None: with self.assertRaises(AssertionError): NamespaceHelper( namespace_str="custom_1::custom_2", entity_name="", max_level=1 )