Nigerian_languages / tests /test_model.py
Gabriel Okiri
Initial commit
4bb9d41
raw
history blame
282 Bytes
import pytest
from app.model.model import NigerianLanguageModel
from app.model.config import ModelConfig
def test_model_initialization():
config = ModelConfig()
model = NigerianLanguageModel(config)
assert model.tokenizer is not None
assert model.model is not None