File size: 282 Bytes
4bb9d41
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
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