You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
789 B
25 lines
789 B
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)
|