25 lines
789 B
Python
25 lines
789 B
Python
|
import pytest
|
||
|
|
from spacy import registry
|
||
|
|
|
||
|
|
FUNCTIONS = [
|
||
|
|
("loggers", "spacy.WandbLogger.v1"),
|
||
|
|
("loggers", "spacy.WandbLogger.v2"),
|
||
|
|
("loggers", "spacy.WandbLogger.v3"),
|
||
|
|
("loggers", "spacy.WandbLogger.v4"),
|
||
|
|
("loggers", "spacy.WandbLogger.v5"),
|
||
|
|
("loggers", "spacy.MLflowLogger.v1"),
|
||
|
|
("loggers", "spacy.MLflowLogger.v2"),
|
||
|
|
("loggers", "spacy.ClearMLLogger.v1"),
|
||
|
|
("loggers", "spacy.ClearMLLogger.v2"),
|
||
|
|
("loggers", "spacy.ChainLogger.v1"),
|
||
|
|
("loggers", "spacy.PyTorchLogger.v1"),
|
||
|
|
("loggers", "spacy.LookupLogger.v1"),
|
||
|
|
("loggers", "spacy.CupyLogger.v1"),
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("reg_name,func_name", FUNCTIONS)
|
||
|
|
def test_registry(reg_name, func_name):
|
||
|
|
assert registry.has(reg_name, func_name)
|
||
|
|
assert registry.get(reg_name, func_name)
|